单薄刺猬

文章
5
资源
0
加入时间
2年10月21天

assert all(tensors[0].size(0) == tensor.size(0) for tensor in tensors) AssertionError

出现这个问题,我总结了两个原因1.Data.TensorDataset()里的两个输入,不是tensor类型我上面的from_numpy是将numpy.narray转换成tensor2.Data.TensorDataset()里的两个输入,维度不一致比如下面的第一行,第一维代表数据的行数。Data.TensorDataset()里的两个输入,应该改成一样的第二行维度一...