使用libquadmath和eigen

时间:2017-05-12 15:28:17

标签: c++ eigen eigen3

我想使用__float128和eigen的erf()函数,但found out目前它只支持浮点数和双精度数:

  

此函数仅支持c ++ 11中的float和double标量类型   模式。支持其他标量类型,或非c ++ 11中的float / double   模式,用户必须为任何标量提供erf(T)的实现   输入T类型。

由于我想使用__float128,如果可能的话,我想依靠libquadmath s erfq implementation。但是怎么做呢? 我目前能想到的唯一(丑陋?)方式是使用eigens unaryExpr()。还有其他可能性吗?

1 个答案:

答案 0 :(得分:3)

你可以专攻Eigen::internal::erf_impl(当然,类似于任何其他功能):

#include <quadmath.h>
#include <iostream>
#include <unsupported/Eigen/SpecialFunctions>

namespace Eigen { namespace internal {
template<>
struct erf_impl<__float128> {
  EIGEN_DEVICE_FUNC
  static EIGEN_STRONG_INLINE __float128 run(__float128 x) { return ::erfq(x); }
};
}}

int main()
{
    typedef Eigen::Array<__float128, Eigen::Dynamic, 1> ArrayXF;
    ArrayXF a(4); a << 0, 0.25, 0.5, 0.75;
    ArrayXF b = a.erf();

    for(int i=0; i<4; ++i){
        char buf[100];
        quadmath_snprintf(buf, 100, "%.50Qe", b[i]); std::cout << buf << '\n';
    }
}

输出:

  

0.00000000000000000000000000000000000000000000000000e + 00   2.76326390168236932985068267764815703534647315720851e-01   5.20499877813046537682746653891964513119913394193564e-01   7.11155633653515131598937834591410814324096358715387e-01