nnlib
GPU-accelerated, C/C++ neural network library.
Public Member Functions | List of all members
Function< Types > Class Template Referenceabstract
Inheritance diagram for Function< Types >:
BackwardFunction

Public Member Functions

sTensor forward (const Types &... args)
 
std::vector< sTensor > backward (sTensor grad) override
 
virtual sTensor forwardFn (const Types &... args)=0
 
virtual std::vector< sTensor > backwardFn (sTensor grad)=0
 

Additional Inherited Members

- Public Attributes inherited from BackwardFunction
std::vector< sTensor > parents
 

The documentation for this class was generated from the following file: