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()裏的兩個輸入，應該改成一樣的 第二行維度一樣，就不會出現問

>>阅读原文<<