Python中List、numpy.array与torch.Tensor 之间的相互转化

list 与 numpy

①list → numpy

array = np.array(list)

②numpy → list

list = array.tolist()

torch.Tensor 与 numpy

①torch.Tensor → numpy

array = tensor.numpy()

②numpy → list

tensor = torch.from_numpy(array)

list 与 torch.Tensor

①list → tensor

tensor = torch.Tensor(list)

②tensor →numpy → list

list = tensor.numpy().tolist()