到此，就完成了对PyTorch算子的学习。如果你要学习其他算子，可以先从aten/src/ATen/native目录的相关函数入手，从native_functions.yaml中找到dispatch目标函数，详情可以参考Figure 1。
更多精彩文章，欢迎扫码关注下方的公众号, 并访问我的简书博客：https://www.jianshu.com/u/c0fe8671254e
欢迎转发至朋友圈，工作号转载请后台留言申请授权～
深入浅出PyTorch（算子篇） (2)
内容版权声明：除非注明，否则皆为本站原创文章。