output with shape [1, 28, 28] doesn't match the broadcast shape [3, 28, 28]
output with shape [1, 28, 28] doesn’t match the broadcast shape [3, 28, 28]
下载MINST数据集,并查看一个批次的图像时,出现了如上的错误
transform = transforms.Compose([
transforms.ToTensor(),
transforms.Normalize(mean=[0...
原创
2019-10-24 18:56:28 ·
1714 阅读 ·
0 评论