Nd4j INDArray 从布尔值到 int

Question

如果我有这个数组（它是一个 INDArray，但我正在简化视图） myArray = [[0, 0.7], [0, 0]]，我这样做 res = myArray.eq(0.7) 我得到res = [[false, true], [false, false]]

是否有可以将布尔 INDArrays 转换为 0/1 Arrays 的 Nd4j 方法？得到这个[[0, 1], [0, 0]]

Answer 1

在 INDArray 中有可用的 cast 方法，它会为您提供您想要的 - 0 和 1 的数组