如何重新定义“ torch.nn.Conv2d”？我需要添加一个新参数。可以用C ++或C重构它吗？

Question

我设计了一种需要重新定义Conv2d的新算法，我可以通过pytorch的C ++ API重新定义CONV2吗？ CONV2的源代码在哪里？

classtorch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True)

我们需要：

classtorch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True,new_parameter)