[深度学习框架] Pytorch源码学习之Tensor

本文深入探讨PyTorch中的Tensor,揭示其作为Python对象的本质,包括zero-copy操作、数据指针和Tensor Storage。讲解了Tensor如何通过数据指针链接到原始数据,并在CPU和GPU上进行内存分配。同时,提到了与numpy tensor之间的转换及其差异。

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

Tensor

总览:Tensor对象中包含Storage对象,Storage完成对原数据的抽象:Storage中包含指向原始数据的指针以及用于内存分配的Allocator
在这里插入图片描述
pytorch虽然是python first的设计,但所有heavy work都是由C++实现的

在python 中,C++代码的集成由extension完成

pytorch使用基础张量操作库ATen,其他都在ATen基础上建立

Autograd就是在ATen上实现的,它能进行自动微分。

python中,万物皆对象

在这里插入图片描述

zero-copy操作

numpy和pytorch的tensor很方便相互转换,但是torch.tensor(np_array)torch.from_numpy(np_array)对数据的拷贝有些不同
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值