pytorch - pytorch conv2d的源码在哪里？

Question

我在哪里可以找到 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是他们如何导入功能

所以我去了那里，但我没有找到代码。

标签： pytorch

Answer 1

功能代码全部用 C++ 实现。C++ pytorch 代码的入口点conv2d在这里。