Method: Torch::NN::Module#register_parameter

Defined in:
lib/torch/nn/module.rb

#register_parameter(name, param) ⇒ Object 



25
26
27
28
 
# File 'lib/torch/nn/module.rb', line 25

def register_parameter(name, param)
  # TODO add checks
  @parameters[name] = param
end