关键代码
x = torch.randn(64,3,128,128)
transform=transforms.Compose(
[transforms.Resize(64), transforms.ToTensor(), transforms.Normalize([0.5], [0.5])]
)
transform(x)
原因
在于 x 本就是 Tensor 类型的,有写了一次ToTensor()转换类型,因此会报错。
解决办法
删除 transforms.ToTensor()
或者
修改x 类型为其他类型