如何在tensorflow.js中提取张量的值?

时间:2018-11-10 17:29:32

标签: javascript tensorflow tensorflow.js

我的模型输出十个预测,其中最大的预测为argmax。不幸的是,它也返回一个张量(具有单个值,例如:Tensor:[8]),我无法弄清楚如何使该张量成为整数或提取唯一值。

到目前为止,我已经尝试收集asType,dataSync和get。

1 个答案:

答案 0 :(得分:1)

dataSync返回一个数组。要获得单个元素,需要使用以下数组索引

end