| Server IP : 217.160.0.212 / Your IP : 216.73.216.141 Web Server : Apache System : Linux www 6.18.51-i1-ampere #1196 SMP Fri Sep 11 20:43:55 CEST 2026 aarch64 User : sws1073854427 ( 1073854427) PHP Version : 8.4.23 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : OFF | Pkexec : OFF Directory : /lib/python3/dist-packages/pythran/pythonic/include/numpy/ |
Upload File : |
#ifndef PYTHONIC_INCLUDE_NUMPY_SQUARE_HPP
#define PYTHONIC_INCLUDE_NUMPY_SQUARE_HPP
#include "pythonic/include/types/numpy_op_helper.hpp"
#include "pythonic/include/utils/numpy_traits.hpp"
#include "pythonic/include/utils/functor.hpp"
#include <complex>
namespace wrapper
{
}
PYTHONIC_NS_BEGIN
namespace numpy
{
namespace wrapper
{
template <class T>
auto square(T const &arg) -> decltype(arg *arg)
{
return arg * arg;
}
template <class T>
std::complex<T> square(std::complex<T> const &arg)
{
T r = arg.real(), i = arg.imag();
auto t = r * i;
auto r2 = r * r;
auto i2 = i * i;
return {r2 - i2, t + t};
}
}
#define NUMPY_NARY_FUNC_NAME square
#define NUMPY_NARY_FUNC_SYM wrapper::square
#include "pythonic/include/types/numpy_nary_expr.hpp"
}
PYTHONIC_NS_END
#endif