def forward(self, x):
    print(x.shape, self.weight.shape, self.kernel_size, self.stride, self.padding, self.dilation, self.groups)
    output = F.conv2d(x, self.weight, self.bias, self.stride, self.padding, self.dilation, self.groups)
    return output
def forward(self, x):
    print(x.shape, self.weight.shape, self.kernel_size, self.stride, self.padding, self.dilation, self.groups)
    output = F.conv2d(x, self.weight, self.bias, self.stride, self.padding, self.dilation, self.groups)
    return output