org.nd4j.linalg.indexing.NDArrayIndex to org.nd4j.linalg.indexing.INDArrayIndex