Nd4j INDArray от логического до int

Question

Если у меня есть этот массив (это 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]]

Answer 1

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