分类：tensor维度不一样

2021
07-22

解决Pytorch dataloader时报错每个tensor维度不一样的问题

使用pytorch的dataloader报错：RuntimeError:stackexpectseachtensortobeequalsize,butgot[2]atentry0and[1]atentry11.问题描述报错定位:位于定义dataset的代码中def__getitem__(self,index):...returny#此处报错报错内容File"D:\python\lib\site-packages\torch\utils\data\_utils\collate.py",line55,indefault_collatereturntorch.stack(batch,0,out=out)RuntimeError:stackexpectseac... 继续阅读 >