NumCpp  2.7.0
A Templatized Header Only C++ Implementation of the Python NumPy Library
nth_root.hpp
Go to the documentation of this file.
1 #pragma once
29 
30 #include "NumCpp/NdArray.hpp"
33 #include "NumCpp/Utils/powerf.hpp"
34 
35 namespace nc
36 {
37  //============================================================================
38  // Method Description:
45  template<typename dtype1, typename dtype2>
46  double nth_root(dtype1 inValue, dtype2 inRoot) noexcept
47  {
50 
51  return utils::powerf(static_cast<double>(inValue), 1.0 / static_cast<double>(inRoot));
52  }
53 
54  //============================================================================
55  // Method Description:
62  template<typename dtype1, typename dtype2>
63  NdArray<double> nth_root(const NdArray<dtype1>& inArray, dtype2 inRoot)
64  {
65  NdArray<double> returnArray(inArray.shape());
66  stl_algorithms::transform(inArray.cbegin(), inArray.cend(), returnArray.begin(),
67  [inRoot](dtype1 inValue) noexcept -> double
68  {
69  return nth_root(inValue, inRoot);
70  });
71 
72  return returnArray;
73  }
74 } // namespace nc
#define STATIC_ASSERT_ARITHMETIC(dtype)
Definition: StaticAsserts.hpp:37
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
OutputIt transform(InputIt first, InputIt last, OutputIt destination, UnaryOperation unaryFunction)
Definition: StlAlgorithms.hpp:702
auto powerf(dtype1 inValue, const dtype2 inPower) noexcept
Definition: Utils/powerf.hpp:49
Definition: Coordinate.hpp:45
double nth_root(dtype1 inValue, dtype2 inRoot) noexcept
Definition: nth_root.hpp:46