Class: LossResult

Inherits:
Object
  • Object
show all
Defined in:
lib/toy/models/transformer.rb

Overview

Output of cross_entropy_grad: dL/dlogits matrix + scalar loss.

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(dlogits, loss) ⇒ LossResult

Returns a new instance of LossResult.



413
# File 'lib/toy/models/transformer.rb', line 413

def initialize(dlogits, loss); @dlogits = dlogits; @loss = loss; end

Instance Attribute Details

#dlogitsObject

Returns the value of attribute dlogits.



412
413
414
# File 'lib/toy/models/transformer.rb', line 412

def dlogits
  @dlogits
end

#lossObject

Returns the value of attribute loss.



412
413
414
# File 'lib/toy/models/transformer.rb', line 412

def loss
  @loss
end