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 |

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 |

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

- /home/jenkins-mlpack/mlpack.org/_src/mlpack-3.2.1/src/mlpack/methods/ann/visitor/weight_size_visitor.hpp