Corsdfgfadgfg 2022-04-10 19:40 采纳率: 100%
浏览 268
已结题

permute(2, 0, 1)在pytorch的tensor处理中有什么用吗

在对数据集进行处理的时候经常看到permute(2, 0, 1)这个操作,为啥不是1,2,0这种,2,0,1这个顺序会带来什么好处吗?

  • 写回答

1条回答 默认 最新

  • 对象被抛出 2022-04-10 20:11
    关注

    交换3个维度的顺序, 假设原来的顺序是(W, H, B), permute(2, 0, 1)之后就是(B, W, H), 因为torch里面第一个维度普遍都是batchnum,

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论

报告相同问题?

问题事件

  • 系统已结题 4月19日
  • 已采纳回答 4月11日
  • 创建了问题 4月10日