我在哪里可以找到pytorch函数conv2d的源代码？
它应该在torch.nn.functional中，但如果我搜索conv2d，我只找到_add_docstr行。
https://github.com/pytorch/pytorch/blob/master/torch/nn/functional.py
更新：这不是我输入的，我指的是函数。Conv2d类使用了nn.functional中的conv2d函数
此处：
https://github.com/pytorch/pytorch/blob/master/torch/nn/modules/conv.py
第338行：
return F.conv2d(F.pad(input, expanded_padding, mode='circular')F是他们如何导入函数
所以我去了那里，但我没有找到密码。
1条答案
按热度按时间h43kikqp1#
函数代码是用C实现的。从1.13.1版开始，
conv2d的C代码的入口点是aten/src/ATen/native/Convolution.cpp：804。如果您对函数如何注册到API更感兴趣，那么您可以查看aten/src/ATen/native/README.md。更深入地了解PyTorch中的一些设计决策将非常有益。（see here）。更多一般信息可以在PyTorch developer's wiki中找到，不过请记住，这个wiki主要是供贡献者使用的工具，并不像Python API文档那样精致。IMO一个很好的起点是Core Frontend Onboarding页面，它提供了了解PyTorch源代码所需的大多数内容的链接。