python3　list, np.array, torch.tensor相互转换

ndarray = np.array(list)# list 转 numpy数组list = ndarray.tolist()# numpy 转 listtensor=torch.Tensor(list)# list 转 torch.Tensorlist = tensor.numpy().tolist()# torch.Tensor 转 list先转numpy，后转listndarray = te

关切得大神

515人浏览 · 2021-04-28 21:32:05

关切得大神 · 2021-04-28 21:32:05 发布

ndarray = np.array(list)  # list 转 numpy数组
list = ndarray.tolist()  # numpy 转 list
tensor=torch.Tensor(list)  # list 转 torch.Tensor
list = tensor.numpy().tolist()  # torch.Tensor 转 list  先转numpy，后转list
ndarray = tensor.cpu().numpy()  # torch.Tensor 转 numpy  *gpu上的tensor不能直接转为numpy
tensor = torch.from_numpy(ndarray)  # numpy 转 torch.Tensor