[pytorch报错]TypeError: tensor is not a torch image.

最新推荐文章于 2022-09-23 16:56:57 发布

cmd75

最新推荐文章于 2022-09-23 16:56:57 发布

阅读量2.3k

点赞数 2

文章标签： python pytorch

本文链接：https://blog.csdn.net/kaigeshishen/article/details/103465288

版权

[pytorch报错]TypeError: tensor is not a torch image.

原因是使用torchvision.transforms.Compose传入的参数中Normalize一定要在ToTensor后面
要这么写：

transform=transforms.Compose([
                      transforms.Resize((28, 28)),  
                      transforms.ToTensor(),
                      transforms.Normalize((0.1307,), (0.3081,))]))

而不能：

transform=transforms.Compose([
                      transforms.Resize((28, 28)),  
                      transforms.Normalize((0.1307,), (0.3081,)),
                      transforms.ToTensor()]))