Pythorch-在哪里实施__conv1d__?

1. 它依赖于后端(GPU、CPU、分布式等),但在最有趣的GPU案例中,它是从 cuDNN 它以二进制格式发布,因此您无法检查其源代码。对CPU来说也是一个类似的故事 MKLDNN .我不知道PyTorch会在哪里“手摇”它自己的卷积内核,但我可能是错的。 编辑 :的确,我错了,正如下面的答案所指出的。
2. 不知道pytorch的结构很难。正如所解释的,许多代码实际上是基于各种标记文件自动生成的。 here . 要想弄清楚这一点,需要进行大量的跳跃。例如, conv.cpp 正在链接的文件 uses torch::conv1d ,定义为 here 及用途 at::convolution 反过来使用 at::_convolution 例如,它发送到多个变量 at::cudnn_convolution . At::Cudnn_卷积 我相信是创造出来的 here 通过标记文件直接插入 cuDNN 实现(尽管我不能在代码中准确地指出发生这种情况时的具体点)。

下面是我得到的答案 pytorch discussion board :

我相信这里定义了手卷-ed卷积: https://github.com/pytorch/pytorch/blob/master/aten/src/THNN/generic/SpatialConvolutionMM.c 三

NN模块实现如下: https://github.com/pytorch/pytorch/tree/master/aten/src GPU版本为THCUnn,CPU版本为THnn