PyTorchの関数本体ってどこで定義されてるの？逆伝搬は？~ATenの世界へのみちしるべ~ - Qiita

はじめに 本稿はPyTorch中級者以上向けのニッチな内容ですが，どうしても計算方法の定義が見たいという方にはおすすめです．内容は主に３つです． PythonからC++で書かれたATenライブラリへの橋渡し (conv2dを例に挙げ，)ATen内で順伝搬関数が扱われているかを見る 逆伝搬関数がいかに順伝搬関数と紐づいているかを見る 隠された実装 PyTorchの公式ドキュメントを見ながらプログラミングをし，時には[Source]ボタンを押してgithubの実装を確認したりしますよね．でも[Source]ボタンがないときありますよね？例えばtorch.nn.Conv2dには[Source]ボタンがありますが，torch.nn.functional.conv2dにはありません． ・・・なんで同じConv2dなのに実装が見れないの？ と思いつつ，PyTorchのgithubを確認してみます．to

[misshiki]

“PyTorch中級者以上向け..． PythonからC++で書かれたATenライブラリへの橋渡し、(conv2dを例に挙げ，)ATen内で順伝搬関数が扱われているかを見る、逆伝搬関数がいかに順伝搬関数と紐づいているかを見る”