WeightSizeVisitor Class Reference
WeightSizeVisitor returns the number of weights of the given module. More...
Inheritance diagram for WeightSizeVisitor:
Public Member Functions | |
template < typename LayerType > | |
| size_t | operator() (LayerType *layer) const |
| Return the number of weights. More... | |
Detailed Description
WeightSizeVisitor returns the number of weights of the given module.
Definition at line 27 of file weight_size_visitor.hpp.
Member Function Documentation
◆ operator()()
| size_t operator() | ( | LayerType * | layer | ) | const |
Return the number of weights.
The documentation for this class was generated from the following file:
- /home/ryan/src/mlpack.org/_src/mlpack-3.1.0/src/mlpack/methods/ann/visitor/weight_size_visitor.hpp