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