module Num::Grad
#
Extended modules
Num::Grad
Methods#
#register(name : String, gate : Num::Grad::Gate(U), result : Num::Grad::Variable(U), *parents : Num::Grad::Variable(U)) forall U
#
(name : String, gate : Num::Grad::Gate(U), result : Num::Grad::Variable(U), *parents : Num::Grad::Variable(U)) forall U
Cached a node in the computational graph. This is only required if an operation needs to be backpropogated.
Arguments#
- name :
String- Description of the operation - gate :
Gate- Operation gate containing a backward method and cached arguments - result :
Variable- The result of the operation being cached - parents :
Variable- The operands present in operation being cached
This method should be used sparingly by Users of the application. It should only be necessary when a User is defining their own custom activation function or Layer.