列表转数组: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)
本文介绍了如何在Python中通过NumPy和TensorFlow进行数据类型转换,包括列表转数组、数组转张量等关键操作。特别关注了GPU环境下张量的处理技巧。此外,还涵盖了将张量转换回列表以及使用torch.from_numpy创建张量的方法。

3525

被折叠的 条评论
为什么被折叠?



