What is the NDArray
equivalent for numpy transpose like this: out = np.transpose(in, (0,2,1,3))
?
(in
is of rank 4)
Asked
Active
Viewed 188 times
1 Answers
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
-
Glad to know that you've found out the solution ;) – Shams Ul Azeem Aug 13 '18 at 21:17