如何在TensorFlow.js中获取2D张量数据并保留其形状？

嘿，这个问题我之前玩TensorFlow.js的时候也踩过同款坑！data()方法本身就是设计用来返回张量底层扁平化的一维原始数据的，所以你得到那个Float32Array完全是它的预期行为。如果想要获取和原张量形状一致的数组，这两个方法可以直接解决你的问题：

方法1：使用异步的array()方法 #

这个方法和data()一样是异步的，返回一个Promise，当Promise resolve后，你就能拿到保持原形状的数组了：

const X = tf.tensor2d([[1, 2, 3, 4], [2, 2, 5, 3]]);
X.array().then(arr => console.log(arr));
// 输出结果: [[1, 2, 3, 4], [2, 2, 5, 3]]

方法2：使用同步的arraySync()方法 #

如果你的场景不需要异步操作（比如处理小型张量），可以直接用同步方法获取结果，省去Promise的链式调用：

const X = tf.tensor2d([[1, 2, 3, 4], [2, 2, 5, 3]]);
const arr = X.arraySync();
console.log(arr);
// 输出结果: [[1, 2, 3, 4], [2, 2, 5, 3]]

小提示 #

• 处理大型张量时，优先用异步的array()，避免同步方法阻塞主线程影响页面性能；
• data()方法更适合只需要原始扁平化数据的场景，比如一些底层数值运算或者数据序列化的需求。

内容的提问来源于stack exchange，提问作者hadrienj