typescript - Tensorflow.js `tensor2d` 到数值同时保留形状

Question

在 Angular7 项目中使用 tensorflowjs，我在访问 tensor2d 对象的数值时遇到问题。

确实:

let tensor2d = tf.tensor2d( [[1,2],[3,4]])
let tensor2d_array = tensor2d.dataSync()

将给出形状为 (4,) 的 Array 对象，而不是我希望的 (2,2,)。

我怎样才能保持形状并获得我的张量数组。 (在所有情况下，手动 reshape 并不实用，因为我无法访问此项目中的 js 矩阵库。)。

Answer 1

dataSync和 data返回扁平化 typedArray .这意味着 tensor2d_array 的形状为 [4]。

与其获取与张量具有相同形状的数据，不如仅获取要检索的索引值更实用。话虽如此，如果出于某种原因，您想在获取索引[i, j] 处的值之前先获取数据数组，这answer显示了如何检索使用 data 或 dataSync

获取的数据的元素 [i, j]

更新

从版本 15 开始，可以使用 tensor.array() 获得与张量形状相同的数组或 tensor.arraySync()