【pytorch】torch.tensor-np.array的转换

本文详细介绍了PyTorch tensor与NumPy array之间的转换,包括同内存转换和复制转换,以及如何在CPU与GPU间切换。同内存转换时,改变一方会影响另一方。而复制转换则保持独立。在进行运算时,需确保所有变量在同一设备上。

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

各有两种转换方式,一种是把pytorch.tensor 和对应 numpy放在同一个内存空间,改变tensor时numpy也会改变(The Torch Tensor and NumPy array will share their underlying memory locations (if the Torch Tensor is on CPU), and changing one will change the other.),另一种是单纯的copy,改变一个另一个不会发生变化。

1.同内存转换

1.1 tensor->numpy

我们可以看到改变numpy的值了以后,对应的tensor值也会改变。

>>>x = torch.randn(2,3)
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

学渣渣渣渣渣

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值