rensor与numpy相互转换

原创于 2022-02-19 21:48:52 发布 · 907 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #python #pytorch

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

29 篇文章

订阅专栏

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

部署运行你感兴趣的模型镜像

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

您可能感兴趣的与本文相关的镜像