TypeError: transpose(): argument 'dim0' (position 1) must be int, not tuple
程序员文章站
2024-01-15 15:30:55
...
1. 问题描述
TypeError: transpose(): argument 'dim0' (position 1) must be int, not tuple
如图所示：
2. 原因分析
代码中的 i 是tensor类型的，不是narray类型的，pytorch中的tensor.transpose()只支持两维的张量形变，两维以上可以使用tensor.permute()。
3. 解决办法
ims = [[plt.imshow(i.permute(1,2,0), animated=True)] for i in img_list]
改进后的代码如图：