numpy.transpose() 函数

参考博客：详解numpy中transpose()函数

numpy 和 pytorch 中都有 transpose() 方法