@Test
public void testDimShuffle() {
IComplexNDArray n = Nd4j.complexLinSpace(1, 4, 4).reshape(2, 2);
IComplexNDArray twoOneTwo =
n.dimShuffle(new Object[] {0, 'x', 1}, new int[] {0, 1}, new boolean[] {false, false});
assertTrue(Arrays.equals(new int[] {2, 1, 2}, twoOneTwo.shape()));
IComplexNDArray reverse =
n.dimShuffle(new Object[] {1, 'x', 0}, new int[] {1, 0}, new boolean[] {false, false});
assertTrue(Arrays.equals(new int[] {2, 1, 2}, reverse.shape()));
}
