Nd4j INDArray 从布尔值到整数

Nd4j INDArray from boolean to int

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

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

在 INDArray 中有可用的转换方法,它会给你你想要的 - 0 和 1 的数组