WeightSizeVisitor returns the number of weights of the given module. More...
#include <weight_size_visitor.hpp>
Inheritance diagram for mlpack::ann::WeightSizeVisitor:
Collaboration diagram for mlpack::ann::WeightSizeVisitor:
Public Member Functions | |
| template<typename LayerType > | |
| size_t | operator() (LayerType *layer) const |
| Return the number of weights. More... | |
| size_t | operator() (MoreTypes layer) const |
Detailed Description
WeightSizeVisitor returns the number of weights of the given module.
Member Function Documentation
◆ operator()()
template<typename LayerType >
|
inline |
Return the number of weights.
WeightSizeVisitor visitor class.
The documentation for this class was generated from the following files:
- src/mlpack/methods/ann/visitor/weight_size_visitor.hpp
- src/mlpack/methods/ann/visitor/weight_size_visitor_impl.hpp
