关键代码
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 类型为其他类型
文章出处登录后可见！
已经登录？立即刷新
