pytorch：索引超出批处理大小的索引答案

【问题标题】：pytorch: index out of bound for index bigger than batch sizepytorch：索引超出批处理大小的索引
【发布时间】：2020-03-23 01:12:06
【问题描述】：

我正在尝试使用 Pytorch 从 MNIST 预测一些图像，但我无法选择 0-60,000 之间的任何索引。首先我添加了“drop_last=True”，因为我注意到最后一个不完整的批次没有被丢弃，我认为这可以解决问题，但事实并非如此。如果我的批量大小为 256，我可以选择预测的最大索引为 255。我应该如何解决这个问题？以下是我获取图像的方式：

images = MNIST('mnist_data',transform=T, download=True, train=True)
image_loader = torch.utils.data.DataLoader(images,batch_size=256, drop_last=True, shuffle=True)

以及我如何尝试做出预测：

image_index = 258
value = (images[image_index])
prediction = Net().forward(value)

然后我得到 IndexError：索引 258 超出了尺寸为 256 的维度 0 的范围

【问题讨论】：

标签： mnist index-error batchsize

【解决方案1】：

这主要是因为您试图索引一个大于数组/张量长度的值。你会想要增加批量大小来修复它

【讨论】：