如何将numpyndarray转换到位？

我正在使用numpy.

我有一个形状为[T,H,W,C]的ndarray,我想把它转换为:[T,C,H,W].但是,这个阵列非常庞大,我想要节省内存.

但我发现np.transpose这样做并非就地.

为什么这样的操作np.transpose没有就地对应？

我以前认为任何命名的操作np.Bar都会以其原位对应命名np.Bar_,但却发现这不是事实.

来自np.transposedocs

尽可能返回视图.

意味着没有为输出数组分配额外的内存.

>>> import numpy as np

>>> A = np.random.rand(2, 3, 4, 5)
>>> B = np.transpose(A, axes=(0, 3, 1, 2))

>>> A.shape
(2, 3, 4, 5)
>>> B.shape
(2, 5, 3, 4)

您可以np.shares_memory用来检查是否B是以下视图A:

>>> np.shares_memory(A, B)
True

因此,您可以安全地转置数据np.transpose.