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

出现这个问题，我总结了两个原因

1.Data.TensorDataset()里的两个输入，不是tensor类型

我上面的from_numpy是将numpy.narray转换成tensor

 

2.Data.TensorDataset()里的两个输入,维度不一致

比如下面的第一行，第一维代表数据的行数。Data.TensorDataset()里的两个输入，应该改成一样的

第二行维度一样，就不会出现问题