Python 中列表(list)、数组(np.array)及 张量(tensor)的转换
数据转换
·
列表转数组:np.array(list)
数组转列表:np.array.tolist()
列表转张量:torch.Tensor(list)
张量转列表:先转numpy,后转list,tensor.numpy().tolist()
张量转数组:tensor.numpy()
注:在 gpu 上的张量不能直接转为numpy,需要使用cpu(),tensor.cpu().numpy()
数组转张量:torch.from_numpy(ndarray),tf.conver_to_tensor(ndarray)
更多推荐
已为社区贡献2条内容
所有评论(0)