gpt4 book ai didi

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

转载 作者:行者123 更新时间:2023-12-05 02:13:15 27 4
gpt4 key购买 nike

我使用 model.predict() 通过 tensorflow.js 输出一个张量 A(size:512*512*3) 然后我将它整形为 A.reshape(512*512*3 ).但现在我想将这个张量转换为数组,以便我可以将它与 three.js 一起使用。如何解决这个问题?

最佳答案

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

  • data()dataSync() 有一个扁平的类型数组

但目前支持的类型有float32int32;因此相应的 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 - 如何在 tensorflowjs 中将张量转换为 Uint8Array 数组,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55059475/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com