Nd4J: получить индекс, где существует значение

Question

как получить индекс, где существует определенное значение. В numpy:

import numpy as np
myArr = np.array()
index = np.where(myArr == someValue)
// Output: an index value consisting rows and cols will be given

В ND4J я дошел до этого, но я не знаю, что ввести в параметр условия:

INDArray index = myArr.getWhere(someValue, condition=??);

Другими словами, как найти элемент в INDArray в ND4J?

Answer 1

BooleanIndexing.firstIndex (INDArray, Condition) - это то, что вы ищете, я думаю.

Answer 2

просто используйте Conditions.equals

сначала импортируйте Conditions

import org.nd4j.linalg.indexing.conditions.Conditions;

затем:

myArr.getWhere(someValue, Conditions.equals(1));