Как переопределить «torch.nn.Conv2d»? Мне нужно добавить новый параметр. Могу ли я восстановить его с помощью C ++ или C?

Question

Я разрабатываю новый алгоритм, который должен переопределить Conv2d. Могу ли я переопределить CONV2 через API C ++ pytorch? Где находится исходный код для 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)