Contents

module Torch
  module NN
    class ReLU < Module
      def initialize #(inplace: false)
        # @inplace = inplace
      end

      def forward(input)
        F.relu(input) #, inplace: @inplace)
      end
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
torch-rb-0.1.3	lib/torch/nn/relu.rb
torch-rb-0.1.2	lib/torch/nn/relu.rb
torch-rb-0.1.1	lib/torch/nn/relu.rb
torch-rb-0.1.0	lib/torch/nn/relu.rb