nnlib
GPU-accelerated, C/C++ neural network library.
|
Public Member Functions | |
sTensor | forwardFn (const sTensor &args) override |
std::vector< sTensor > | backwardFn (sTensor grad) override |
Public Member Functions inherited from Function< sTensor > | |
sTensor | forward (const Types &... args) |
std::vector< sTensor > | backward (sTensor grad) override |
virtual sTensor | forwardFn (const Types &... args)=0 |
Private Attributes | |
sTensor | cacheA |
Additional Inherited Members | |
Public Attributes inherited from BackwardFunction | |
std::vector< sTensor > | parents |