RuntimeError: stack expects each tensor to be equal size, but got [x] at entry 0 and [x] at entry 1

问题描述：在生成dataloader的时候，训练集可以跑，测试集出现此错误：RuntimeError: stack expects each tensor to be equal size, but got [200] at entry 0 and [116] at entry 1。

解决思路：在生成dataloader需要生成dataset，那么我的错误发生原因是因为dataset中有一个minibatch的数据个数与其他minibatch不同，遂进入自定义dataset方法中查看，通过打印调试，发现是数据集label的问题。

解决方案：进入dataset中，打印dataset输出的结果。