def forward(self, x):
    avg_out = torch.mean(x, dim=1, keepdim=True)
    (max_out, _) = torch.max(x, dim=1, keepdim=True)
    x = torch.cat([avg_out, max_out], dim=1)
    x = self.conv1(x)
    return self.sigmoid(x)
def forward(self, x):
    avg_out = torch.mean(x, dim=1, keepdim=True)
    (max_out, _) = torch.max(x, dim=1, keepdim=True)
    x = torch.cat([avg_out, max_out], dim=1)
    x = self.conv1(x)
    return self.sigmoid(x)