web-dev-qa-db-ja.com

Tensorflow.jsでテンソル内のアイテムの値を取得します

インデックスを指定した後、Tensorflow.jsのテンソルから値を取得するにはどうすればよいですか？

indexingtensortensorflow.js
8
Pranay Aryal

これにはdatasyncを使用できます。

const newTensor = tf.tensor2d([[2,4],[5,6]]);
const tensorData = newTensor.dataSync();
console.log("data[0] is " + tensorData[0]);
console.log("data[3] is " + tensorData[3]);

https://codepen.io/anon/pen/NMKgeO?editors=1011

10
BlessedKey
const newTensor = tf.tensor2d([[2,4], [5,6]])
newTensor.get([0]) ##returns 2
newTensor.get([3]) ##returns 6

ありがたいことに、これはすべてテンソルではなく数値を返します。 TensorFiddle ここ：）

4
Pranay Aryal

次のより強力な方法を使用できます

tensor.buffer（）。get（0、0）;

これにより、テンソルの論理座標（平坦化された1d座標ではなく2d座標）にインデックスを付けることができます。 リンク を参照してください

1
Raza