ATLAS Offline Software
|
#include <Stack.h>
Public Member Functions | |
Stack () | |
Stack (size_t n_inputs, const std::vector< LayerConfig > &layers, size_t skip_layers=0) | |
~Stack () | |
Stack (Stack &)=delete | |
Stack & | operator= (Stack &)=delete |
VectorXd | compute (VectorXd) const |
size_t | n_outputs () const |
Private Member Functions | |
size_t | add_layers (size_t n_inputs, const LayerConfig &) |
size_t | add_dense_layers (size_t n_inputs, const LayerConfig &) |
size_t | add_normalization_layers (size_t n_inputs, const LayerConfig &) |
size_t | add_highway_layers (size_t n_inputs, const LayerConfig &) |
size_t | add_maxout_layers (size_t n_inputs, const LayerConfig &) |
Private Attributes | |
std::vector< ILayer * > | m_layers |
size_t | m_n_outputs |
lwtDev::Stack::Stack | ( | size_t | n_inputs, |
const std::vector< LayerConfig > & | layers, | ||
size_t | skip_layers = 0 |
||
) |
lwtDev::Stack::~Stack | ( | ) |
|
delete |
|
private |
|
private |
|
private |
|
private |
|
private |
VectorXd lwtDev::Stack::compute | ( | VectorXd | in | ) | const |
size_t lwtDev::Stack::n_outputs | ( | ) | const |