tensor维度转换

一、查看当前维度 .size()

二、张量变形  .view(3,-1)      # -1 是根据维度自己变化

三、解压 unsqueeze() 

  压缩 squeeze(),可指定具体维度,dim=1

四、扩大张量 .expand()

五、重复张量 .repeat(2,3)    # 在指定维度重复原 tensor

六、矩阵转置.t

  必须是二维张量,也就是矩阵才可以用

七、维度置换 torch.transpose(x,1,2)   # 置换1和2

      torch.transpose(x,0,1)  #置换0和1

八、维度指定    permute

原文地址:https://www.cnblogs.com/DoctorZhao/p/13221206.html