javascript - Tensorflow.js tf.Tensor 转 JS 数

Question

我正在使用 Tensorflow.js，我想绘制一些存储为 tf.Tensors 的数据。我找不到将它们转换为普通 JS 数字的方法。

const a = tf.Scalar(1)
const b = tf.Scalar(2)
var array = []
// Now o want to push the numbers 1 and 2 to mye array
array.push(a)
array.push(b)

此代码为我提供了一个张量对象数组。我想要一个数字数组。知道如何进行吗？谢谢。

Answer 1

.dataSync() 下载 GPU 的值并返回 Tensor 中包含的值的 Float32Array。因此，如果您想将值作为张量存储在数组中并希望从中获取一个数字数组，您可以使用 map() 映射它。创建下载值的数组，因为您在示例中使用标量，所以您可以直接映射到 Float32Array 中的第一个元素以获取其值:

let values = array.map(t => t.dataSync()[0])

例子:

const a = tf.scalar(1)
const b = tf.scalar(2)
var array = []
array.push(a)
array.push(b)

let values = array.map(t => t.dataSync()[0])
console.log(values);

<script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@0.11.2">
</script>