def __init__(self, inchannel):
    super().__init__()
    self.conv = torch.nn.Conv2d(inchannel, 1, kernel_size=1, stride=1)
def __init__(self, inchannel):
    super().__init__()
    self.conv = torch.nn.Conv2d(inchannel, 1, kernel_size=1, stride=1)