class OnnxChainer::Operators::Relu

Public Class Methods

new(input_names:, output_names:, instance_variable_name:, need_initialized:) click to toggle source
# File lib/onnx-chainer/operators/relu.rb, line 19
def initialize(input_names:, output_names:, instance_variable_name:, need_initialized:)
  @input_names = input_names
  @output_names = output_names
  @instance_variable_name = instance_variable_name
  @need_initialized = need_initialized
end
parse(node, input_names, inputs, output_name_index) click to toggle source
# File lib/onnx-chainer/operators/relu.rb, line 7
def parse(node, input_names, inputs, output_name_index)
  need_initialized = node.input.any? { |i| inputs.map(&:name).include?(i) }

  output_names = {
    node.output.first => "r#{output_name_index}"
  }
  instance_variable_name = "@r#{output_name_index}"

  self.new(input_names: input_names, output_names: output_names, instance_variable_name: instance_variable_name, need_initialized: need_initialized)
end

Public Instance Methods

chainer_class() click to toggle source
# File lib/onnx-chainer/operators/relu.rb, line 26
def chainer_class
  ::Chainer::Functions::Activation::Relu
end
to_call_string(args) click to toggle source
# File lib/onnx-chainer/operators/relu.rb, line 30
def to_call_string(args)
  "#{@output_names.values.first} = #{chainer_class}.relu(#{args.join(', ')})"
end