探索Pytorch Internals

2024年3月26日 190次阅读来源: 深度碎片

TH 和 THNN （C语言库）是Pytorch的最底层；

读懂pytorch的python层的代码，能帮助我们理解组装积木搭建模型的过程；

但要了解每块积木的特性和内在构造，就需要读懂pytorch的C/C++层的代码，也就是TH，THNN库的源代码；

如何有效的理解TH，THNN，以及pytorch 与它们的链接逻辑？

A quick tour of Torch internals
A Tour of PyTorch Internals (Part I)
PyTorch Internals Part II – The Build System

TH， THNN 解读

href=”http://www.bilibili.com/video/av15221530/“>理解torch.nn.functional.relu的底层THNN.Threshold_updateOutput 代码文档

relu的逻辑是什么？C语言代码是怎么写的？应该如何一句句解读代码？

来自stackoverflow的解救：

    原文作者：深度碎片
    原文地址: https://zhuanlan.zhihu.com/p/29939336
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。