我正在尝试对批量标准化进行研究,并且不得不对 pytorch BN 代码进行一些修改。我深入研究了 pytorch 代码并陷入了困境torch.nn.functional.batch_norm,其中引用了torch.batch_norm.
torch.nn.functional.batch_norm
torch.batch_norm
问题是torch.batch_norm无法在火炬库中进一步找到。有什么办法可以找到这个内置函数的源代码并重新实现它吗?谢谢!
pytorch
pytorch ×1