pytorch torch.from_numpy接口

本文详细介绍了如何使用PyTorch的torch.from_numpy()函数从NumPy数组创建张量,强调了数据共享机制及操作限制。通过示例展示了修改张量如何影响原始NumPy数组。

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

pytorch torch.from_numpy


功能说明:从numpy数组创建一个张量,数组和张量共享相同内存.

torch.from_numpy(ndarray) → Tensor
Creates a Tensor from a numpy.ndarray.
从numpy.ndarray创建一个张量。

The returned tensor and ndarray share the same memory. Modifications to the tensor will be reflected in the ndarray and vice versa. The returned tensor is not resizable.
返回的张量和ndarray共享相同的内存。
对张量的修改将反映在ndarray中,反之亦然。 返回的张量不可调整大小。

Example:

>>> a = numpy.array([1, 2, 3])
>>> t = torch.from_numpy(a)
>>> t
tensor([ 1,  2,  3])
>>> t[0] = -1
>>> a
array([-1,  2,  3])

尊重原创:
https://blog.csdn.net/Dontla/article/details/105844900

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值