Class: RubyZero::NN::Layers::ModelStack

Inherits:
ModelList show all
Defined in:
lib/rubyzero/nn/layers/modelstack.rb

Overview

The stack of layers This class don’t supports to containing different Model classes.

Direct Known Subclasses

MultiLayerPerceptron

Instance Method Summary collapse

Methods inherited from ModelList

#+, #__get_str__, #each, #elements, #initialize, #parameters

Methods inherited from Model

#__get_str__, #call, #eval, #initialize, #inspect, load, #parameters, #save, #train

Constructor Details

This class inherits a constructor from RubyZero::NN::Layers::ModelList

Instance Method Details

#<<(layer) ⇒ Object 



5
6
7
8
9
10
11
12
 
# File 'lib/rubyzero/nn/layers/modelstack.rb', line 5

def <<(layer)
    if @models.size > 0
        if layer.class != @models.last.class
            raise TypeError, "Layer type mismatch"
        end
    end
    super(layer)
end

#forward(*args) ⇒ Object 



13
14
15
16
17
18
 
# File 'lib/rubyzero/nn/layers/modelstack.rb', line 13

def forward(*args)
    @models.each do |model|
        args = model.forward(*args)
    end
    return args
end