def __init__(self, n_input, n_output, kernel_size=3, stride=2):
    super().__init__()
    self.c1 = torch.nn.ConvTranspose2d(n_input, n_output, kernel_size=kernel_size, padding=(kernel_size // 2), stride=stride, output_padding=1)
def __init__(self, n_input, n_output, kernel_size=3, stride=2):
    super().__init__()
    self.c1 = torch.nn.ConvTranspose2d(n_input, n_output, kernel_size=kernel_size, padding=(kernel_size // 2), stride=stride, output_padding=1)