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 的数组
如果我有这个数组(它是一个 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 的数组