Pytorch源码
hellopipu
这个作者很懒,什么都没留下…
展开
-
[深度学习框架] Pytorch源码学习之Tensor
Tensor 总览:Tensor对象中包含Storage对象,Storage完成对原数据的抽象:Storage中包含指向原始数据的指针以及用于内存分配的Allocator pytorch虽然是python first的设计,但所有heavy work都是由C++实现的 在python 中,C++代码的集成由extension完成 pytorch使用基础张量操作库ATen,其他都在ATen基础上建...原创 2019-04-04 11:30:00 · 1172 阅读 · 0 评论 -
[深度学习框架] Pytorch源码学习之JIT
JIT–Just-In-Time Compiler JIT有两种模式,一种叫tracing,一种叫scripting tracing不会记录任何控制流信息,如if条件句和循环。但是scripting会保留这些信息。tracing和scripting都包含在TorchScript中,TorchScript的主要目的是从python代码中串行化和优化模型,用TorchScript实现的代码能较方便地...原创 2019-04-04 15:05:30 · 2470 阅读 · 1 评论 -
[深度学习框架] Pytorch源码学习之源码基础一
代码段一 // Python object that backs torch.autograd.Variable struct THPVariable { PyObject_HEAD torch::autograd::Variable cdata; PyObject* backward_hooks; }; PyObject_HEAD这个宏用于标准化python对象,其拓展...原创 2019-04-04 21:34:04 · 2129 阅读 · 0 评论