问题

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的重要性有了一定的了解。