mlpack 3.4.2
Public Member Functions | List of all members
WeightSizeVisitor Class Reference

WeightSizeVisitor returns the number of weights of the given module. More...

#include <weight_size_visitor.hpp>

Inheritance diagram for WeightSizeVisitor:
Inheritance graph
[legend]

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

◆ operator()() [1/2]

size_t operator() ( LayerType *  layer) const

Return the number of weights.

◆ operator()() [2/2]

size_t operator() ( MoreTypes  layer) const

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