pytorch conv2d的源代码在哪里？

Question

我在哪里可以找到pytorch函数conv2d的源代码？

它应该在 torch.nn.functional 中，但如果我搜索 conv2d，我只能找到 _add_docstr 行。我在这里看了：

更新：这不是我的打字，我的意思是功能。Conv2d 类使用 nn.functional 中的 conv2d 函数

这里：

在第 338 行：

return F.conv2d(F.pad(input, expanded_padding, mode='circular')

F 是他们如何导入功能

然后我去了那里，但我没有找到代码。

Answer 1

功能代码全部用C++实现。该入口点的C ++代码pytorchconv2d是在这里。