pytorch报错： invalid argument 0: Sizes of tensors must match except in dimension 0.

这个问题是在dataloader遇到的，网上一般是数据的维度不同，但我这里其实是最后一份数据和之前的数据比数量不足，所以凑不够一个batch，所以我的做法是加入drop_last=True意思是如果数据不够就舍弃最后一个batch。