📌  相关文章
📜  Tensorflow.js tf.Tensor 类 .array() 方法(1)

📅  最后修改于: 2023-12-03 15:05:33.390000             🧑  作者: Mango

Tensorflow.js tf.Tensor 类 .array() 方法

Tensorflow.js 是 Google 开发的一个基于 JavaScript 的机器学习库，可以在浏览器或 Node.js 中使用。tf.Tensor 类是 Tensorflow.js 中的一个重要类，表示一个多维数组的张量，并提供了许多张量的操作方法。其中，.array() 方法可以将张量转化为 JavaScript 的数组。

语法
tf.Tensor.array()
返回值

返回一个包含张量数据的 JavaScript 数组。

示例
const tensor = tf.tensor([1, 2, 3]);
const arr = tensor.array();
console.log(arr); // [1, 2, 3]
注意事项

  • 这个方法只对 CPU 张量有效。

  • 如果张量不是一个浮点数类型，会先将其转换为浮点数再输出。

  • 如果张量太大，将会导致内存溢出的风险。因此，使用该方法应当谨慎考虑是否需要将张量转为数组。

  • 该方法是同步调用的，因此在执行期间可能会阻塞其它的代码。如果想在后台中处理数据，请使用异步 API，如 .data() 方法。

参考资料