If your goal is to do transpose convolution, why don’t you use nn.functional.conv_transpose2d? torch.nn.functional — PyTorch 1.8.1 documentation
If your goal is to do transpose convolution, why don’t you use nn.functional.conv_transpose2d? torch.nn.functional — PyTorch 1.8.1 documentation