如何部分转置一个张量

Question

比如一个n×128×9的ndarray或者张量，怎么把它变成n×9×128

Answer 1

可以使用numpy.transpose()函数进行部分转置，如：

ndarray_transposed = numpy.transpose(ndarray, (0,2,1))

ndarray_transposed是ndarray的转置，其中（0,2,1）表示把ndarray的维度顺序变换为（0,2,1），即ndarray的第0维变为第0维，第2维变为第1维，第1维变为第2维。