RuntimeError: The size of tensor a (0) must match the size of tensor b (31) at non-singleton dimension 0

这个错误信息表示，您正在尝试将一个形状为 的张量和一个形状为 [31] 的张量执行某种操作，但它们在第 0 个维度上的大小不匹配。换句话说，这两个张量的第一个维度的元素数量不同，因此无法执行所需的操作。您需要检查您的代码，并确保这两个张量的形状匹配，或者使用适当的方法来处理它们的形状以满足需求。可能的原因是您的数据集中有一些样本的大小不同，导致在进行批处理时出现了这个错误。您可以通过将参数drop_last设置为True来解决这个问题，这将在最后一个批次中包含不足batch_size的样本。另外，您还可以使用torch.utils.data.sampler.RandomSampler来随机采样数据集，以确保每个批次中的样本大小相同。