首页 > 其他 > 详细

tensor维度转换

时间:2020-07-01 20:44:28      阅读:135      评论:0      收藏:0      [点我收藏+]

一、查看当前维度 .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

 

tensor维度转换

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

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!