class DNN::Layers::MergeLayer
Public Class Methods
call(x1, x2, *args)
click to toggle source
# File lib/dnn/core/layers/merge_layers.rb, line 5 def self.call(x1, x2, *args) new(*args).call(x1, x2) end
Public Instance Methods
call(input1, input2)
click to toggle source
# File lib/dnn/core/layers/merge_layers.rb, line 9 def call(input1, input2) input1 = Tensor.convert(input1) if !input1.is_a?(Tensor) && !input1.is_a?(Param) input2 = Tensor.convert(input2) if !input2.is_a?(Tensor) && !input2.is_a?(Param) if input1.data.is_a?(Xumo::NArray) build(input1.data.shape[1..-1]) unless built? else build([1]) unless built? end forward(input1, input2) end