tensorflow - 访问 TensorFlow.js 中的 tensor2d 的单个值

Question

我正在尝试使用 TensorFlow.js 包访问二维矩阵第二行的元素。下面我使用 [0] 和 [1] 访问矩阵第一行的第一个和第二个元素，但老实说，我希望访问这些元素例如 [0][0] 表示获取 1，[1][1] 表示获取 4。

const t1 = tf.tensor2d([
                         [1,2],
                         [3,4]
                         ]);

    const tensorData = t1.dataSync();
    console.log("data[0] is " + tensorData[0]); // I get 1
    console.log("data[1] is " + tensorData[1]); // I get 2
    console.log("data[0][0] is " + tensorData[0][0]); // I get undefined, but I would expect to get 1

有人可以解释一下为什么吗？我怎样才能访问矩阵第二行的单个元素？

Answer 1

如果你想通过坐标获取张量值，可以使用get方法。这是一个例子

const a=tf.tensor([1, 2, 3, 4], [2, 2])
b=a.get(0,0)
console.log(b)

您还可以查看以下answer