contrib/python/pythran/pythran/pythonic/include/numpy/random/binomial.hpp


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

#ifndef PYTHONIC_INCLUDE_NUMPY_RANDOM_BINOMIAL_HPP
#define PYTHONIC_INCLUDE_NUMPY_RANDOM_BINOMIAL_HPP

#include "pythonic/include/types/ndarray.hpp"
#include "pythonic/types/numpy_expr.hpp"

#include "pythonic/include/numpy/random/generator.hpp"

PYTHONIC_NS_BEGIN
namespace numpy
{
  namespace random
  {
    template <class pS>
    types::ndarray<long, pS> binomial(double n, double p, pS const &shape);

    auto binomial(double n, double p, long size)
        -> decltype(binomial(n, p, types::array<long, 1>{{size}}));

    long binomial(double n, double p, types::none_type d = types::none_type());

    DEFINE_FUNCTOR(pythonic::numpy::random, binomial);
  }
}
PYTHONIC_NS_END

#endif