NumCpp  2.7.0
A Templatized Header Only C++ Implementation of the Python NumPy Library
polygamma.hpp
Go to the documentation of this file.
1 #pragma once
29 
30 #ifndef NUMCPP_NO_USE_BOOST
31 
34 #include "NumCpp/NdArray.hpp"
35 
36 #include "boost/math/special_functions/polygamma.hpp"
37 
38 namespace nc
39 {
40  namespace special
41  {
42  //============================================================================
43  // Method Description:
52  template<typename dtype>
53  auto polygamma(uint32 n, dtype inValue)
54  {
56 
57  return boost::math::polygamma(n, inValue);
58  }
59 
60  //============================================================================
61  // Method Description:
70  template<typename dtype>
71  auto polygamma(uint32 n, const NdArray<dtype>& inArray)
72  {
73  NdArray<decltype(polygamma(n, dtype{0}))> returnArray(inArray.shape());
74 
75  stl_algorithms::transform(inArray.cbegin(), inArray.cend(), returnArray.begin(),
76  [n](dtype inValue) -> auto
77  {
78  return polygamma(n, inValue);
79  });
80 
81  return returnArray;
82  }
83  } // namespace special
84 } // namespace nc
85 
86 #endif // #ifndef NUMCPP_NO_USE_BOOST
#define STATIC_ASSERT_ARITHMETIC(dtype)
Definition: StaticAsserts.hpp:37
Holds 1D and 2D arrays, the main work horse of the NumCpp library.
Definition: NdArrayCore.hpp:72
const_iterator cbegin() const noexcept
Definition: NdArrayCore.hpp:1216
Shape shape() const noexcept
Definition: NdArrayCore.hpp:4283
const_iterator cend() const noexcept
Definition: NdArrayCore.hpp:1524
iterator begin() noexcept
Definition: NdArrayCore.hpp:1166
auto polygamma(uint32 n, dtype inValue)
Definition: polygamma.hpp:53
auto polygamma(uint32 n, const NdArray< dtype > &inArray)
Definition: polygamma.hpp:71
OutputIt transform(InputIt first, InputIt last, OutputIt destination, UnaryOperation unaryFunction)
Definition: StlAlgorithms.hpp:702
Definition: Coordinate.hpp:45
std::uint32_t uint32
Definition: Types.hpp:40