blob: 784b017d7e378693b0f6d91e24d5ab40744b88fc (
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
|
#ifndef PYTHONIC_INCLUDE_NUMPY_ONESLIKE_HPP
#define PYTHONIC_INCLUDE_NUMPY_ONESLIKE_HPP
#include "pythonic/include/numpy/ones.hpp"
#include "pythonic/include/utils/functor.hpp"
PYTHONIC_NS_BEGIN
namespace numpy
{
template <class E, class dtype>
auto ones_like(E const &expr, dtype d = dtype())
-> decltype(ones(sutils::getshape(expr), d));
template <class E>
auto ones_like(E const &expr, types::none_type d = builtins::None)
-> decltype(ones(sutils::getshape(expr),
types::dtype_t<typename E::dtype>()));
DEFINE_FUNCTOR(pythonic::numpy, ones_like)
} // namespace numpy
PYTHONIC_NS_END
#endif
|
