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