如何拆分多维numpy数组

Question

我需要将160000x32x32x2 numpy数组拆分为两个160000x32x32x1 numpy数组。显然，我可以使用for循环来这样做，但是由于大小，这需要很长时间。我正在寻找一种更快的方法来完成此操作，但无法在线找到任何内容。