Class PReLULayer.Builder

    • Constructor Detail

      • Builder

        public Builder()
    • Method Detail

      • inputShape

        public PReLULayer.Builder inputShape​(long... shape)
        Explicitly set input shape of incoming activations so that parameters can be initialized properly. This explicitly excludes the mini-batch dimension.
        Parameters:
        shape - shape of input data
      • sharedAxes

        public PReLULayer.Builder sharedAxes​(long... axes)
        Set the broadcasting axes of PReLU's alpha parameter. For instance, given input data of shape [mb, channels, height, width], setting axes to [2,3] will set alpha to shape [channels, 1, 1] and broadcast alpha across height and width dimensions of each channel.
        Parameters:
        axes - shared/broadcasting axes
        Returns:
        Builder