Contents

module Torch
  module NN
    class ELU < Module
      def initialize(alpha: 1, inplace: false)
        super()
        @alpha = alpha
        @inplace = inplace
      end

      def forward(input)
        F.elu(input, alpha: @alpha, inplace: @inplace)
      end

      def extra_inspect
        inplace_str = @inplace ? ", inplace: true" : ""
        format("alpha: %s", @alpha) + inplace_str
      end
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
torch-rb-0.18.0	lib/torch/nn/elu.rb
torch-rb-0.17.1	lib/torch/nn/elu.rb
torch-rb-0.17.0	lib/torch/nn/elu.rb
torch-rb-0.16.0	lib/torch/nn/elu.rb