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...
size_t	operator() (MoreTypes layer) const

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

size_t operator()

(

LayerType *

layer

)

const

Return the number of weights.

size_t operator()

(

MoreTypes

layer

)

const

---

The documentation for this class was generated from the following file:

• src/mlpack/methods/ann/visitor/weight_size_visitor.hpp