获取Nd4j数组的任意片段

Question

我想在任意大小的Nd4j中执行切片，就像我可以使用Numpy一样。

a = numpy.arange(100)
a[25:50]

nd4j slice方法只接受维度和索引参数，而不接受长度。我如何才能做到这一点？

Answer 1

我知道这是一个古老的问题，但我在谷歌上搜索这个问题时遇到了这个问题。

通过检查slice的源代码，我认为它只能返回完整的行/列，而不是从索引到索引的部分。为此，您可以使用带有NDArrayIndex实例参数的get方法。例如，此代码是您的numpy代码的翻译。

import org.nd4j.linalg.api.ndarray.INDArray;
import static org.nd4j.linalg.factory.Nd4j.linspace;
import static org.nd4j.linalg.indexing.NDArrayIndex.interval;

class SliceExample {
    public static void main(String[] args) {
        INDArray a = linspace(0, 99, 100);    // up to 99 inclusive
        INDArray s = a.get(interval(25, 50)); // up to 50th non inclusive
    }
}

NDArrayIndex documentation