如何在nd4j中反转矩阵列的元素?

yrefmtwq  于 2021-07-06  发布在  Java
关注(0)|答案(1)|浏览(333)

我正试图用nd4j库用java重写这行python代码。

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

它反转列的元素(换句话说,行的顺序)我使用了这一行,但显然它只在end>begin时起作用

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

我提前感谢你的帮助

aij0ehis

aij0ehis1#

我以一种非常恶劣的方式解决了我的问题:))我要和你分享它。不过,如果你能给出一个有力的答案,我将不胜感激。

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

相关问题