Nd4j INDArray从boolean到int

Question

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

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

Answer 1

在INDArray中，可以使用强制转换方法，它将为您提供所需的确切内容-0和1的数组