如何反转 ND4J 中矩阵列的元素？

Question

我正在尝试使用 ND4j 库重写 JAVA 中的这一行 python 代码。

data[index_odd] = tmp[::-1,:]

它反转列的元素。（换句话说，行的顺序）我使用了这条线，但显然它只在 end>begin

时有效

INDArrayIndex rev_idx = NDArrayIndex.interval(10, 1);

提前感谢您的帮助

Answer 1

我用一种非常讨厌的方式解决了我的问题:)) 我将与您分享。不过，如果你能给出一个可靠的答案，我将不胜感激。

int[] idx = new int[(int) OBJ_imag.shape()[0]];
for(int i = 0 ; i<OBJ_imag.shape()[0] ; i ++)
    idx[i] = (int) (OBJ_imag.shape()[0] - i -1);
imag.put(indx, OBJ_imag.getRows(idx));

如何反转 ND4J 中矩阵列的元素？

how can I reverse the elements of a column of a matrix in ND4J?

java

dl4j

nd4j