nnlib
GPU-accelerated, C/C++ neural network library.
|
Public Member Functions | |
sTensor | forwardFn (const sTensor &a) override |
std::vector< sTensor > | backwardFn (sTensor grad) override |
![]() | |
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 | |
![]() | |
std::vector< sTensor > | parents |