class DNN::Models::FixedModel
Attributes
layers[R]
Public Class Methods
new(output_tensor, layers)
click to toggle source
Calls superclass method
DNN::Models::Model::new
# File lib/dnn/core/models.rb, line 731 def initialize(output_tensor, layers) super() @input_link = get_input_link(output_tensor.link) @layers = layers end
Public Instance Methods
forward(input_tensors)
click to toggle source
# File lib/dnn/core/models.rb, line 737 def forward(input_tensors) if input_tensors.is_a?(Array) input_tensors.each do |tensor| @input_link.forward(tensor) end else @input_link.forward(input_tensors) end end
Private Instance Methods
get_input_link(last_link)
click to toggle source
# File lib/dnn/core/models.rb, line 749 def get_input_link(last_link) get_input_link = -> link do if link.is_a?(Link) return link unless link.prev get_input_link.(link.prev) else return link unless link.prev1 get_input_link.(link.prev1) end end get_input_link.(last_link) end