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()裏的兩個輸入,應該改成一樣的 第二行維度一樣,就不會出現問
相關文章
相關標籤/搜索