小编Sto*_*ree的帖子

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

我正在尝试对批量标准化进行研究,并且不得不对 pytorch BN 代码进行一些修改。我深入研究了 pytorch 代码并陷入了困境torch.nn.functional.batch_norm,其中引用了torch.batch_norm.

问题是torch.batch_norm无法在火炬库中进一步找到。有什么办法可以找到这个内置函数的源代码并重新实现它吗?谢谢!

pytorch

3
推荐指数
1
解决办法
1302
查看次数

标签 统计

pytorch ×1