如何将元素级整数幂与特征一起使用

Question

我想使用特征幂函数获取双精度数组和整数数组的元素明智幂。

下面是使用 Eigen v3.3.4 和 v3.3.7 重现该问题的示例代码：

#include <Eigen/Dense>
int main() {
Eigen::ArrayXd x(10);
Eigen::ArrayXd res(10);
Eigen::ArrayXi exponents(10);
x = Eigen::ArrayXd::Random(10);
exponents = Eigen::ArrayXi::LinSpaced(10, 0, 9);
res = Eigen::pow(x, exponents);
return (0);
}

错误消息很长，但本质上我得到

YOU_MIXED_DIFFERENT_NUMERIC_TYPES__YOU_NEED_TO_USE_THE_CAST_METHOD_OF_MATRIXBASE_TO_CAST_NUMERIC_TYPES_EXPLICITLY

在这种情况下，这对我来说似乎不合适，以及

Eigen3/Eigen/src/Core/functors/BinaryFunctors.h:294:84: error: no type named ‘ReturnType’ in ‘struct Eigen::ScalarBinaryOpTraits<double, int, Eigen::internal::scalar_pow_op<double, int> >’
typedef typename ScalarBinaryOpTraits<Scalar,Exponent,scalar_pow_op>::ReturnType result_type;

Answer 1

如错误消息所示，不能隐式混合标量类型。您必须显式强制转换，以便类型匹配：

res = Eigen::pow(x, exponents.cast<double>());

至于整数类型的专用化，幂函数的模板(作为函子)是：

template<typename ScalarX,typename ScalarY, bool IsInteger = 
NumTraits<ScalarX>::IsInteger&&NumTraits<ScalarY>::IsInteger>

并调用简单pow(x,y)除非两种类型都是整数(IsInteger)，在这种情况下有不同的特化。

数组还有一个常量幂的重载，这似乎不是你要找的。在这种情况下(除非 ggael 纠正我)，你绝对可以实现你自己的CustomBinaryOp