0
点赞
收藏
分享

微信扫一扫

batch_size的探索


问题

test_loader 中的y 表示每一个batch对应的128张图片对应的数字,torch.Size([256])表示什么意思?

batch_size的探索_python

batch_size的探索_机器学习_02

方法

在打印了X的长度之后,发现X的长度也为256,这表示此处用作测试的X是由256个1x28x28的矩阵构成的多元组矩阵集合。也即,y的长度为256,而不是128。

batch_size的探索_tensorflow_03


for X, y in test_loader:  
print(X.shape, y.shape)
print(y)
print(len(X))
print(X)

原因:

在初次设置test_loader的batch_size为256,而不是128.

batch_size的探索_深度学习_04

结语

在本次探索中,通过print(),我对test_loader中的(X, y)的数据格式有了一定的认识,同时对batch_size的重要性有了一定的了解。

举报

相关推荐

0 条评论