Question

我使用model.predict()通过tensorflow.js输出张量A（大小：512 * 512 * 3），然后将其重塑为A.reshape（512 * 512 * 3）。但是现在我想将此张量转换为数组，以便可以与three.js一起使用。如何解决这个问题？

Answer 1

要将张量转换为数组，可以使用

data()或dataSync()具有扁平化的typedarray

但是目前支持的类型为float32，int32；因此，相应的typedArray将为Float32Array和Int32Array。 typedarray构造函数可用于更改typedarray的类型

a = tf.tensor([1, 2, 3, 4])

buffer = a.dataSync().buffer

console.log(new Uint8Array(buffer))

console.log(new Uint16Array(buffer))

console.log(new Float32Array(buffer))

// To retrieve easily uint8 type, one can cast the tensor to `int32`

a = tf.tensor([1, 2, 3, 4], undefined, 'int32')

console.log(a.dtype)

buffer = a.dataSync().buffer
console.log(new Uint8Array(buffer))
console.log(new Float32Array(buffer))

<html>
  <head>
    <!-- Load TensorFlow.js -->
    <script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs@1.0.0"> </script>
  </head>

  <body>
  </body>
</html>

如何在tensorflow.js中将张量转换为Uint8Array数组

1 个答案: