如何在pytorch中查找内置函数源代码

我试图研究批处理规范化，并且不得不对pytorch BN代码进行一些修改。我深入研究了pytorch代码，并被引用的

torch.nn.functional.batch\u norm

卡住了

---

问题是在torch库中无法进一步找到批处理规范。有没有办法找到这个内置函数的源代码并重新实现它？谢谢

它就在那里，但在Python中没有定义。它们在C++中定义在<代码AtEng/< /Cord>目录> < /P> 对于CPU，实现（其中之一，取决于输入是否连续）如下所示：

---

对于CUDA，实现如下：

谢谢，它完美地解决了我的问题。虽然我还是很好奇你是怎么知道密码在哪里的。它们之间有映射吗？我不知道映射。我只是在GitHub上搜索了存储库中的

batch\u norm

，并手动查看了结果。我一直关注着 Atdie文件夹，因为这是C++源的大部分所在。