torch常用操作-torch.transpose、torch.permute、np.transpose
在使用Pytorch中我们经常会对矩阵进行操作,其中比较常用的就是调整维度,交换不同的轴。比如常见的有torch.transpose、torch.permute、np.transpose它们之间是有区别和联系的。以上的函数都可以进行轴转换操作。np.transpose 同 torch.permute一样,一次修改多个轴;torch.transpose一次只能交换2个轴。
原创
2023-02-08 20:23:45 ·
841 阅读 ·
0 评论