java - 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));