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=??);

换句话说，如何在ND4J中查找INDArray中的元素？

Answer 1

我想BooleanIndexing.firstIndex(INDArray，Condition)就是你要找的东西。

Answer 2

只需使用Conditions.equals

第一次导入Conditions

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

然后：

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