Tenncor
|
Go to the source code of this file.
Classes | |
struct | layr::ULayerBuilder |
Builder implementation for activation layer. More... | |
struct | layr::ULayer |
Layer implementation to apply activation and pooling functions. More... | |
Namespaces | |
layr | |
Typedefs | |
using | layr::UnaryF = std::function< NodeptrT(const ULayer &, NodeptrT)> |
Function that takes corresponding unary layer and node. More... | |
using | layr::UnaryptrT = std::shared_ptr< ULayer > |
Smart pointer of unary layer. More... | |
Functions | |
NodeptrT | layr::softmax_from_layer (const ULayer &layer, NodeptrT input) |
NodeptrT | layr::maxpool_from_layer (const ULayer &layer, NodeptrT input) |
NodeptrT | layr::meanpool_from_layer (const ULayer &layer, NodeptrT input) |
UnaryptrT | layr::sigmoid (void) |
Return activation layer using sigmoid. More... | |
UnaryptrT | layr::tanh (void) |
Return activation layer using tanh. More... | |
UnaryptrT | layr::softmax (teq::RankT dim) |
Return activation layer using softmax of specified dimension. More... | |
UnaryptrT | layr::maxpool2d (std::pair< teq::DimT, teq::DimT > dims={0, 1}) |
Return pooling layer using max aggregation. More... | |
UnaryptrT | layr::meanpool2d (std::pair< teq::DimT, teq::DimT > dims={0, 1}) |
Return pooling layer using mean aggregation. More... | |
Variables | |
const std::string | layr::sigmoid_layer_key |
Identifier for sigmoid activation layer. More... | |
const std::string | layr::tanh_layer_key |
Identifier for tanh activation layer. More... | |
const std::string | layr::softmax_layer_key |
Identifier for softmax activation layer. More... | |
const std::string | layr::maxpool2d_layer_key |
Identifier for max pooling layer. More... | |
const std::string | layr::meanpool2d_layer_key |
Identifier for mean pooling layer. More... | |
const std::unordered_map< std::string, UnaryF > | layr::unaries |
Map unary layer identifier to connection function. More... | |