输出Tensorflow.js中一个大张量的所有值



我有一个[174,48]维张量,我想输出所有(而不是以类似于该值的方式将它们压缩到浏览器中的开发人员控制台中。我如何实现这一点?

示例

const tensor = tf.tensor([[1, 2], [3, 4]]);
console.log(JSON.stringify(tensor.arraySync())); // [[1,2],[3,4]]

CCD_ 2创建张量的嵌套数组。然后,JSON.stringify用于在打印到控制台之前将数组转换为字符串。

替代方案

或者,您可以使用tensor.dataSync(),它以单个维度而不是嵌套数组返回数据。

这两个函数都有一个异步版本,不会阻塞UI,这对大型张量特别有用:

  • tensor.array()
  • tensor.data()

最新更新