#include <net_layer.h>
|
bool | Load (const apollo::prediction::LayerParameter &layer_pb) override |
| Load the parameter from a pb message. More...
|
|
void | Run (const std::vector< Eigen::MatrixXf > &inputs, Eigen::MatrixXf *output) override |
| Compute the layer output from inputs. More...
|
|
| Layer ()=default |
| Constructor. More...
|
|
virtual | ~Layer ()=default |
| Destructor. More...
|
|
virtual void | ResetState () |
| Reset the internal state of a layer such as LSTM, GRU. More...
|
|
virtual void | SetState (const std::vector< Eigen::MatrixXf > &states) |
| Set the internal state of a layer. More...
|
|
virtual void | State (std::vector< Eigen::MatrixXf > *states) const |
| Access to the internal state of a layer. More...
|
|
std::string | Name () const |
| Name of a layer. More...
|
|
int | OrderNumber () const |
| Order number of a layer in a network. More...
|
|
◆ Load()
bool apollo::prediction::network::BatchNormalization::Load |
( |
const apollo::prediction::LayerParameter & |
layer_pb | ) |
|
|
overridevirtual |
Load the parameter from a pb message.
- Parameters
-
A | pb message contains the parameters |
- Returns
- True is loaded successively, otherwise False
Reimplemented from apollo::prediction::network::Layer.
◆ Run()
void apollo::prediction::network::BatchNormalization::Run |
( |
const std::vector< Eigen::MatrixXf > & |
inputs, |
|
|
Eigen::MatrixXf * |
output |
|
) |
| |
|
overridevirtual |
The documentation for this class was generated from the following file: