Python 如何随机洗牌的深度，高度，宽度的5D矩阵？

我有一个5D矩阵，其中D是深度，H是高度，W是宽度。例如，矩阵的大小为

16 x 3 x 256 x 128 x 64

，我想随机洗牌最后三个维度（

D，H，W

）以获得新矩阵。我如何用python实现它？输出矩阵如下所示

16 x 3 x 256 x 128 x 64
16 x 3 x 64x 128 x 256
16 x 3 x 256 x 64x 128
16 x 3 x 128x 64x 256
...

---

这将转换最后三个轴，假设有5个轴

new_ix = list(np.random.permutation([2, 3, 4]))
np.transpose(x, [0, 1] + new_ix)

---

你能给我们看一下你当前的代码吗（不用洗牌）？我只需要输入5D矩阵。