get_numpy_type_char.hpp

Go to the documentation of this file.

#ifndef MLPACK_BINDINGS_PYTHON_GET_NUMPY_TYPE_CHAR_HPP
#define MLPACK_BINDINGS_PYTHON_GET_NUMPY_TYPE_CHAR_HPP

#include <mlpack/prereqs.hpp>

namespace mlpack {
namespace bindings {
namespace python {

template<typename T>
inline std::string GetNumpyTypeChar()
{
  return "?";
}

// size_t = s.
template<>
inline std::string GetNumpyTypeChar<arma::Mat<size_t>>()
{
  return "s";
}

template<>
inline std::string GetNumpyTypeChar<arma::Col<size_t>>()
{
  return "s";
}

template<>
inline std::string GetNumpyTypeChar<arma::Row<size_t>>()
{
  return "s";
}

// double = d.
template<>
inline std::string GetNumpyTypeChar<arma::mat>()
{
  return "d";
}

template<>
inline std::string GetNumpyTypeChar<arma::vec>()
{
  return "d";
}

template<>
inline std::string GetNumpyTypeChar<arma::rowvec>()
{
  return "d";
}

} // namespace python
} // namespace bindings
} // namespace mlpack

#endif