blob: a154d9f7a8ea0a798906b8e823d985f14c48f8f0 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
#ifndef PYTHONIC_NUMPY_ATLEAST1D_HPP
#define PYTHONIC_NUMPY_ATLEAST1D_HPP
#include "pythonic/include/numpy/atleast_1d.hpp"
#include "pythonic/numpy/asarray.hpp"
PYTHONIC_NS_BEGIN
namespace numpy
{
template <class T>
typename std::enable_if<
types::is_dtype<T>::value,
types::ndarray<T, types::pshape<std::integral_constant<long, 1>>>>::type
atleast_1d(T t)
{
return {types::pshape<std::integral_constant<long, 1>>(), t};
}
template <class T>
auto atleast_1d(T const &t) ->
typename std::enable_if<!(types::is_dtype<T>::value),
decltype(asarray(t))>::type
{
return asarray(t);
}
}
PYTHONIC_NS_END
#endif
|
