python 中的 matlab permute(A, [3 2 1]) 的等价物是什么?

what is an equivalent of matlab permute(A, [3 2 1]) in python?

如果 A 是一个 2x2 数组,python 中与 MATLAB 中的 permute(A, [3 2 1]) 等效的表达式是什么?

谢谢

您正在寻找numpy.transpose

np.transpose( np.expand_dims(A, axis=2), (2, 1, 0) )

由于 numpy 默认情况下没有尾随单例维度,您需要使用 np.expand_dims

显式添加它

否则 np.expand_dims(A, axis=2) 的 shorthand 是 A[:, :, None] 所以

np.transpose(A[:, :, None], (2,1,0))