check_input_shape.hpp

Go to the documentation of this file.

#ifndef MLPACK_METHODS_ANN_UTIL_CHECK_INPUT_SHAPE_HPP
#define MLPACK_METHODS_ANN_UTIL_CHECK_INPUT_SHAPE_HPP

#include <mlpack/methods/ann/visitor/input_shape_visitor.hpp>

namespace mlpack {
namespace ann {

template<typename T>
void CheckInputShape(const T& network,
                     const size_t inputShape,
                     const std::string& functionName)
{
  for (size_t l = 0; l < network.size(); ++l)
  {
    size_t layerInShape = boost::apply_visitor(InShapeVisitor(), network[l]);
    if (layerInShape == 0)
    {
      continue;
    }
    else if (layerInShape == inputShape)
    {
      break;
    }
    else
    {
      std::string estr = functionName + ": the first layer of the network " +
          "expects " + std::to_string(layerInShape) + " elements, but the " +
          "input has " + std::to_string(inputShape) + " dimensions!";
      throw std::logic_error(estr);
    }
  }
}

} // namespace ann
} // namespace mlpack

#endif