pytorch实现numpy与tensor之间的转换

程序媛罢了

于 2020-04-22 11:43:28 发布

阅读量3.2k

点赞数

分类专栏： python学习记录

本文链接：https://blog.youkuaiyun.com/qq_43688965/article/details/105678159

版权

python学习记录专栏收录该内容

7 篇文章

订阅专栏

本文详细介绍了PyTorch的Tensor对象与NumPy数组之间的相互转换方法，包括从Tensor转换为NumPy数组及从NumPy数组转换为Tensor的具体步骤与示例代码，帮助读者理解并掌握两种数据结构的灵活运用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.tensor转numpy:

a = torch.ones(2,2)b = a.numpy()c=np.array(a) #也可以转numpy数组print(type(a))print(type(b))print(a)print(b)

结果<class ‘torch.Tensor’><class ‘numpy.ndarray’>tensor([[1., 1.],[1., 1.]])[[1. 1.][1. 1.]]

2.numpy转tensor：

2345678910import torchimport numpy as np a = np.ones(5)b = torch.from_numpy(a)c=torch.Tensor(a) #也可以转pytorch Tensorprint(type(a))print(type(b))print(a)print(b)

输出：
<class ‘numpy.ndarray’><class ‘torch.Tensor’>[1. 1. 1. 1. 1.]tensor([1., 1., 1., 1., 1.], dtype=torch.float64)

pytorch的tensor对象与numpy数组是可以相互转换的，且numpy数组的默认类型是double