gpt4 book ai didi

pytorch - pytorch conv2d的源代码在哪里?

转载 作者:行者123 更新时间:2023-12-03 16:17:32 30 4
gpt4 key购买 nike

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

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

最佳答案

功能代码全部用C++实现。 conv2d 的 C++ pytorch 代码的入口点是 here .

关于pytorch - pytorch conv2d的源代码在哪里?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/59127007/

30 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com