因此,该类中的tensor 第一维度必须相等。 from torch.utils.data import TensorDataset. import torch. from torch.utils.data import DataLoader. a = ...
確定! 回上一頁