#ifndef PY_VECTORIZE_HPP

#define PY_VECTORIZE_HPP

#include <type_traits>

#include "pyarray.hpp"

#include "xtensor/xvectorize.hpp"

namespace xt

{

template <class Func, class R, class... Args>

struct pyvectorizer

};

/**

template <class R, class... Args>

inline pyvectorizer<R (*)(Args...), R, Args...> pyvectorize(R (*f)(Args...))

{

return pyvectorizer<R (*)(Args...), R, Args...>(f);

}

/// @cond DOXYGEN_INCLUDE_OVERLOADS

template <class F, class R, class... Args>

inline pyvectorizer<F, R, Args...> pyvectorize(F&& f, R (*)(Args...))

{

return pyvectorizer<F, R, Args...>(std::forward<F>(f));

}

template <class F>

inline auto pyvectorize(F&& f) -> decltype(pyvectorize(std::forward<F>(f), (detail::get_function_type<F>*)nullptr))

{

return pyvectorize(std::forward<F>(f), (detail::get_function_type<F>*)nullptr);

}

/// @endcond

}

#endif