Pytorch中List、tensor、numpy相互转换

1

ndarray = np.array(list)

1

list = ndarray.tolist()

1

tensor=torch.Tensor(list)

1
2

# 先转numpy，后转list
list = tensor.numpy().tolist()

1
2
3

ndarray = tensor.numpy()
# *gpu上的tensor不能直接转为numpy,应先放回cpu中
ndarray = tensor.cpu().numpy()

1

tensor = torch.from_numpy(ndarray)