Class: AdamStepResult

Inherits:
Object
  • Object
show all
Defined in:
lib/toy/ffi/tinynn.rb,
lib/toy/ffi/tinynn_cuda.rb,
lib/toy/ffi/tinynn_metal.rb

Overview

Same AdamStepResult definition as lib/toy/ffi/tinynn.rb and lib/toy/ffi/tinynn_cuda.rb — drivers require exactly one of tinynn_cuda, tinynn_metal so we duplicate to keep each file self-sufficient.

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(param, mom_m, mom_v) ⇒ AdamStepResult

Returns a new instance of AdamStepResult.



326
327
328
329
330
 
# File 'lib/toy/ffi/tinynn.rb', line 326

def initialize(param, mom_m, mom_v)
  @param = param
  @mom_m = mom_m
  @mom_v = mom_v
end

Instance Attribute Details

#mom_m ⇒ Object

Returns the value of attribute mom_m.



325
326
327
 
# File 'lib/toy/ffi/tinynn.rb', line 325

def mom_m
  @mom_m
end

#mom_v ⇒ Object

Returns the value of attribute mom_v.



325
326
327
 
# File 'lib/toy/ffi/tinynn.rb', line 325

def mom_v
  @mom_v
end

#param ⇒ Object

Returns the value of attribute param.



325
326
327
 
# File 'lib/toy/ffi/tinynn.rb', line 325

def param
  @param
end