如何在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));