rensor与numpy相互转换

李划水员

于 2022-02-19 21:48:52 发布

阅读量889

点赞数

CC 4.0 BY-SA版权

分类专栏：小白代码打卡文章标签：深度学习 python pytorch

本文链接：https://blog.youkuaiyun.com/Daisy4/article/details/123024536

小白代码打卡专栏收录该内容

29 篇文章

订阅专栏

本文介绍了如何在PyTorch与NumPy之间进行数据转换。通过torch.ones创建一个张量，然后使用`.numpy()`将其转换为NumPy数组。反之，用numpy数组初始化一个张量，即使修改NumPy数组，PyTorch张量也会同步更新，展示了两者之间的紧密联系。注意，只有非CUDA张量才能转换为NumPy。

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

tensor转numpy： x.numpy()

a = torch.ones(5)
print(a)     ##输出：tensor([1., 1., 1., 1., 1.])

转换：
b = a.numpy()
print(b)     ##输出：[1. 1. 1. 1. 1.]

numpy转tensor： torch.from_numpy(x)

import numpy as np
a = np.ones(5)
b = torch.from_numpy(a)
np.add(a, 1, out=a)
print(a)   ##[1. 1. 1. 1. 1.]
print(b)   ##tensor([1., 1., 1., 1., 1.], dtype=torch.float64)

注：除chartensor外所有tensor都可以转换为numpy