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()里的两个输入,应该改成一样的 第二行维度一样,就不会出现问
相关文章
相关标签/搜索