tensorflow - 如何将映射函数应用于张量？（张量流.js）

Question

如何将函数应用于张量的每个元素，例如

var new_tensor = old_tensor.map(map_function)

标签： tensorflowtensortensorflow.js

Answer 1

如果你使用你可以得到一个你可以映射的dataSync方法。oldTensorTypedArray

var old_tensor_vals = old_tensor.dataSync()
var new_tensor_vals = old_tensor_vals.map(map_function)