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