Nd4j INDArray из логического в 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]]


arrays nd4j
person plguillou    schedule 13.01.2020    source источник

Ответы (1)


arrow_upward
0
arrow_downward

В INDArray доступен метод приведения, он даст вам именно то, что вы хотите - массив из 0 и 1

person raver119    schedule 14.01.2020
comment
Понял, это res = res.castTo(DataType.INT) - person plguillou; 25.01.2020