在哪里可以找到pytorch函数conv2d的源代码?
它应该在torch.nn.functional中,但是我只找到_add_docstr行, 如果我搜索conv2d。我看了这里:
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是他们导入功能的方式
所以我去了那里,但是我没有找到代码。