nnlib
GPU-accelerated, C/C++ neural network library.
Public Member Functions
|
List of all members
Function< Types > Class Template Reference
abstract
Inheritance diagram for Function< Types >:
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:
include/
functions.h
Generated by
1.9.1