0

What is the NDArray equivalent for numpy transpose like this: out = np.transpose(in, (0,2,1,3))?
(in is of rank 4)

tzolov
  • 469
  • 4
  • 6

1 Answers1

0

Apparently the INDArray.permute(dimension rages) does exactly what i've been looking for.

The ND4J's outNDArray = inNDArray.permute(0,2,1,3) is equivalent to NumPy's out = np.transpose(in, (0,2,1,3))

tzolov
  • 469
  • 4
  • 6