segment_softmax

    description: 'layer segment_softmax for configurable component'
    ---

    # segment_softmax layer

    `segment_softmax` is a custom layer that allows for masking multiple output channels.

    Suppose you have 30 channels and `segment_softmax 3`. First, the 30 channels split into 15/15.
    The first 15 will be used for softmax and multiplied against the second.
    Then each channel is softmaxed, multiplied, and summed.

    So 30 input channels with 3 output channels equate to 5 input channels for each output channel.

    ## input size

    Any 4-d tensor of the shape `[B, C, H, W]`

    ## output size

    [B, OUTPUT_CHANNELS, H, W]

    ## syntax

    ```json
      "segment_softmax OUTPUT_CHANNELS"
    ```

    ## examples

    At the end of the generator for RGB images:

    ```json
      "conv 30",
      "segment_softmax 3",
      "hardtanh"
    ```

Last updated