mlpack 3.4.2
|
WeightSizeVisitor returns the number of weights of the given module. More...
#include <weight_size_visitor.hpp>
Public Member Functions | |
template<typename LayerType > | |
size_t | operator() (LayerType *layer) const |
Return the number of weights. More... | |
size_t | operator() (MoreTypes layer) const |
WeightSizeVisitor returns the number of weights of the given module.
Definition at line 27 of file weight_size_visitor.hpp.
size_t operator() | ( | LayerType * | layer | ) | const |
Return the number of weights.
size_t operator() | ( | MoreTypes | layer | ) | const |