Pytorch中的维度变换,reshape、permute、view
程序中常常会涉及到张量的维度变换,为区别常用变换函数reshape、permute、view的不同,使用以下代码直观说明。
先对二维张量进行变换:
import torch
c=torch.randn(2,3)
print("c:")
print(c)
c1=torch.reshape(c,(3,2))
print("reshape:")
print(c1)
c2=c.permute(1,0)
p...
原创
2019-12-01 13:55:21 ·
3241 阅读 ·
0 评论