Class: Informers::Seq2SeqLMOutput

Inherits:
ModelOutput show all
Defined in:
lib/informers/models.rb

Instance Method Summary collapse

Methods inherited from ModelOutput

#[]

Constructor Details

#initialize(logits, past_key_values, encoder_outputs, decoder_attentions = nil, cross_attentions = nil) ⇒ Seq2SeqLMOutput

Returns a new instance of Seq2SeqLMOutput.



1447
1448
1449
1450
1451
1452
1453
1454
 
# File 'lib/informers/models.rb', line 1447

def initialize(logits, past_key_values, encoder_outputs, decoder_attentions = nil, cross_attentions = nil)
  super()
  @logits = logits
  @past_key_values = past_key_values
  @encoder_outputs = encoder_outputs
  @decoder_attentions = decoder_attentions
  @cross_attentions = cross_attentions
end