module Torch
  module NN
    class PReLU < Module
      def initialize(num_parameters: 1, init: 0.25)
        @num_parameters = num_parameters
        super()
        @weight = Parameter.new(Tensor.new(num_parameters).fill!(init))
      end

      def forward(input)
        F.prelu(input, @weight)
      end

      def extra_inspect
        format("num_parameters: %s", @num_parameters)
      end
    end
  end
end