tensor和numpy维度转换

花生吃花生

于 2022-08-25 20:07:52 发布

阅读量536

点赞数

本文链接：https://blog.youkuaiyun.com/m0_59967951/article/details/126531628

版权

深度学习专栏收录该内容

10 篇文章

订阅专栏

本文介绍了如何在NumPy和PyTorch Tensor之间进行高效的数据转换。包括使用torch.from_numpy()将NumPy数组转换为共享相同内存的Tensor，以及如何通过调用Tensor的numpy()方法将Tensor转换回NumPy数组，保持数据的一致性和共享内存。

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

1. numpy转tensor（共享内存）

使用torch.from_numpy() 将numpy转为tensor

import numpy as np
import torch
n = np.random.randn(224, 224, 3)
print(n.shape)
t = torch.from_numpy(n)
print(t.shape)

(224, 224, 3)
torch.Size([224, 224, 3])

2.tensor转numpy（共享内存）

使用.numpy()转为numpy

a = torch.randn(3,4,5)
print(a.shape)
b = a.numpy()
print(b.shape)

torch.Size([3, 4, 5])
(3, 4, 5)