a = complex_random_sample((9, 10))
a.shape
a[2, :]
a[2, :].shape
a[:, 6]
a[:, 6].shape
a[:, 6].reshape(3, 3)
a.reshape(3, 3, a.shape[-1])[:, :, 6]
assert_allclose(a.reshape(3, 3, a.shape[-1])[:, :, 6], a[:, 6].reshape(3, 3))
Try changing any of the 6 in the above line to 7