【问题标题】:Where is the source code of pytorch conv2d?pytorch conv2d的源码在哪里?
【发布时间】:2020-03-26 08:42:24
【问题描述】:

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

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

【问题讨论】:

  • 为什么投反对票?
  • 对于那些投票结束的人,因为“它正在寻求推荐书籍、软件库或其他非现场资源。这个问题可能会导致基于意见的答案。 "这个问题在任何意义上都不是在寻求建议(它是在请求有关图书馆的特定信息)。这个问题很清楚,当然不可能导致基于意见的答案,所以请不要因为这个原因试图关闭它。

标签: pytorch


【解决方案1】:

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

【讨论】:

    猜你喜欢
    • 2018-06-21
    • 1970-01-01
    • 2022-11-19
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2011-04-20
    • 1970-01-01
    相关资源
    最近更新 更多