#include <DenseLayerImp.hpp>
◆ DenseLayerImp()
Constructor ingesting a TensoTwoD and TensorOneD object.
- Parameters
-
[in] | weights | The TensorTwoD object containing weights |
[in] | biases | The TensorOneD object containing biases |
- Exceptions
-
geopm::Exception | if the weights is empty |
geopm::Exception | if the dimensions are incompatible, i.e. if weights rows is inequal to bias dimension. |
◆ forward()
Inference step.
- Parameters
-
- Exceptions
-
geopm::Exception | if input vector is incompatible with the DenseLayer |
- Returns
- Returns a TensorOneD object of output values
Implements geopm::DenseLayer.
◆ get_input_dim()
size_t geopm::DenseLayerImp::get_input_dim |
( |
| ) |
const |
|
overridevirtual |
Get the dimension required for the input TensorOneD.
- Returns
- Returns a size_t equal to the number of columns of weights
Implements geopm::DenseLayer.
◆ get_output_dim()
size_t geopm::DenseLayerImp::get_output_dim |
( |
| ) |
const |
|
overridevirtual |
Get the dimension of the resulting TensorOneD.
- Returns
- Returns a size_t equal to the number of rows of weights
Implements geopm::DenseLayer.
The documentation for this class was generated from the following files: