def forward(self, x):
    x = self.relu(x)
    out = torch.cat([self.conv1(x), self.conv2(x[:, :, 1:, 1:])], dim=1)
    out = self.bn(out)
    return out
def forward(self, x):
    x = self.relu(x)
    out = torch.cat([self.conv1(x), self.conv2(x[:, :, 1:, 1:])], dim=1)
    out = self.bn(out)
    return out