def __init__(self, in_channels: int):
        """
        Args:
            in_channels: The number of channels
                in the feature map of the input.
        """
        super().__init__()
        self.conv = nn.Conv2d(in_channels, 1, kernel_size=1, stride=1)

        Args:
            in_channels: The number of channels
                in the feature map of the input.
        
catalyst/contrib/layers/se.py