Poplar and PopLibs
|
Parameters to describe a fully-connected layer. More...
#include <FullyConnected.hpp>
Public Attributes | |
std::size_t | numGroups |
The number of groups (where each group represents a fully-connected layer of the same shape) that operate in one layer. More... | |
std::size_t | batchSize |
Number of samples in the input to the layer. | |
std::size_t | inputSize |
Size of the input in each batch into the layer. | |
std::size_t | outputSize |
Size of the output in each batch from the layer. | |
Parameters to describe a fully-connected layer.
std::size_t poplin::fc::FullyConnectedParams::numGroups |
The number of groups (where each group represents a fully-connected layer of the same shape) that operate in one layer.
Each group is totally independent of the others and so numGroups
is a common dimension among the inputs, weights, and outputs for the layer.