Inheritance diagram for Matmul:
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 |
Private Attributes | |
| sTensor | cacheA |
| sTensor | cacheB |
Additional Inherited Members | |
| Public Attributes inherited from BackwardFunction | |
| std::vector< sTensor > | parents |
The documentation for this class was generated from the following files:
- include/functions.h
- src/math/functions.cpp
