mlpack
|
InShapeVisitor returns the input shape a Layer expects. More...
#include <input_shape_visitor.hpp>
Public Member Functions | |
template<typename LayerType > | |
size_t | operator() (LayerType *layer) const |
Return the input shape of layer. | |
size_t | operator() (MoreTypes layer) const |
template<typename LayerType > | |
std::size_t | operator() (LayerType *layer) const |
InShapeVisitor visitor class. | |
InShapeVisitor returns the input shape a Layer expects.