在哪里可以找到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是他们如何导入功能
所以我去了那里,但是我没有找到代码。
功能代码全部用C ++实现。该入口点的C ++代码pytorchconv2d
是在这里。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句