在对数据集进行处理的时候经常看到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,
本回答被题主选为最佳回答 , 对您是否有帮助呢?解决 1无用
在对数据集进行处理的时候经常看到permute(2, 0, 1)这个操作,为啥不是1,2,0这种,2,0,1这个顺序会带来什么好处吗?
交换3个维度的顺序, 假设原来的顺序是(W, H, B), permute(2, 0, 1)之后就是(B, W, H), 因为torch里面第一个维度普遍都是batchnum,