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

Public Member Functions

sTensor forwardFn (const sTensor &a, const sTensor &b) override
 
std::vector< sTensor > backwardFn (sTensor grad) override
 
- Public Member Functions inherited from Function< sTensor, sTensor >
sTensor forward (const Types &... args)
 
std::vector< sTensor > backward (sTensor grad) override
 
virtual sTensor forwardFn (const Types &... args)=0
 

Additional Inherited Members

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

Detailed Description

Examples
MNIST, and Titanic.

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