|
OpenNN
Open-source neural networks library
|
Workspace holding the activations of every layer during a forward pass. More...
#include <forward_propagation.h>
Public Member Functions | |
| ForwardPropagation (const Index=0, NeuralNetwork *=nullptr) | |
| Constructs a workspace for the given batch size and network. | |
| void | set (const Index=0, NeuralNetwork *=nullptr) |
| Reconfigures the workspace for a new batch size or network; reuses allocations when possible. | |
| TensorView | get_last_trainable_layer_outputs () const |
| Returns the output tensor of the last trainable layer. | |
| TensorView | get_outputs () const |
| Returns the final output tensor of the network. | |
| void | print () const |
| Prints a human-readable summary of the workspace contents. | |
Public Attributes | |
| Index | batch_size = 0 |
| NeuralNetwork * | neural_network = nullptr |
| Buffer | data |
| vector< vector< vector< TensorView > > > | views |
Workspace holding the activations of every layer during a forward pass.
| opennn::ForwardPropagation::ForwardPropagation | ( | const Index | = 0, |
| NeuralNetwork * | = nullptr ) |
Constructs a workspace for the given batch size and network.
| batch_size | Maximum number of samples per forward pass. |
| neural_network | Network whose layer specs drive buffer sizing (non-owning). |
| TensorView opennn::ForwardPropagation::get_last_trainable_layer_outputs | ( | ) | const |
Returns the output tensor of the last trainable layer.
| TensorView opennn::ForwardPropagation::get_outputs | ( | ) | const |
Returns the final output tensor of the network.
| void opennn::ForwardPropagation::print | ( | ) | const |
Prints a human-readable summary of the workspace contents.
| void opennn::ForwardPropagation::set | ( | const Index | = 0, |
| NeuralNetwork * | = nullptr ) |
Reconfigures the workspace for a new batch size or network; reuses allocations when possible.
| batch_size | Maximum number of samples per forward pass. |
| neural_network | Network whose layer specs drive buffer sizing (non-owning). |
| Index opennn::ForwardPropagation::batch_size = 0 |
| Buffer opennn::ForwardPropagation::data |
| NeuralNetwork* opennn::ForwardPropagation::neural_network = nullptr |
| vector<vector<vector<TensorView> > > opennn::ForwardPropagation::views |