PyTorch 源码浅析(四)

自动微分的算符库THNN和THCUNN 在不考虑自动微分的引擎的情况下,实际上想要实现简单的自动微分很简单。只要将不同的算符实现为具有forward方法,和bac…

PyTorch 源码浅析(三)

THC——基于CUDA的张量代数库 THC里实际上已经开始使用C++了,并不是一个像TH一样的纯C的库。这可能也是个人更偏爱mshadow的原因,有些地方写得像C有些地…

PyTorch 源码浅析(二)

TensorApply宏 如同作者注释所言,tensor apply系列的宏的机制如下 从最外部的角标开始,循环至第一个发生内存不连续的地址,然后将其记为张量A…

PyTorch 源码浅析(一)

CPU上的张量(多维数组)库 TH库的实现使用了用C语言的宏产生的泛型,并且通过命名规则来产生类似面向对象的效果。这部分我们在这一章后面介绍。 TH…

PyTorch 源码分析:Python 层

尝试使用了pytorch,相比其他深度学习框架,pytorch显得简洁易懂。花时间读了部分源码,主要结合简单例子带着问题阅读,不涉及源码中C拓展库的实现。

Pytorch 源码与运行原理浅析 – 网络篇

记得当时刚刚接触的时候一直搞不明白,为什么自己只是定义了几个网络,就可以完整的训练整个模型,它背后的机制又是如何,搞明白了这个,才有可能去做更…

如何有效地阅读 PyTorch 的源代码?

本文作者罗秀哲,来源知乎。 PyTorch的模块化感觉很好,读某一个部分不需要很清楚其它部分的具体实现。不过复数是大部分文件里都需要检查一下要…

返回顶部