问题
test_loader 中的y 表示每一个batch对应的128张图片对应的数字，torch.Size([256])表示什么意思？
方法
在打印了X的长度之后，发现X的长度也为256，这表示此处用作测试的X是由256个1x28x28的矩阵构成的多元组矩阵集合。也即，y的长度为256，而不是128。
for X, y in test_loader: print(X.shape, y.shape) print(y) print(len(X)) print(X) 原因： 在初次设置test_loader的batch_size为256，而不是128.
|
结语
在本次探索中，通过print()，我对test_loader中的（X, y）的数据格式有了一定的认识，同时对batch_size的重要性有了一定的了解。