在 Eigen 中混合标量类型

Question

#include <iostream>

#include <Eigen/Core>

namespace Eigen {

// float op double -> double
template <typename BinaryOp>
struct ScalarBinaryOpTraits<float, double, BinaryOp> {
  enum { Defined = 1 };
  typedef double ReturnType;
};

// double op float -> double
template <typename BinaryOp>
struct ScalarBinaryOpTraits<double, float, BinaryOp> {
  enum { Defined = 1 };
  typedef double ReturnType;
};

}

int main() {
    Eigen::Matrix<float, Eigen::Dynamic, Eigen::Dynamic> m1(2, 2);
    m1 << 1, 2, 3, 4;

    Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic> m2(2, 2);
    m2 << 1, 2, 3, 4;

    std::cerr << m1 * m2 <<std::endl;  // <- boom!!
}

我想知道为什么上面的代码不能编译。 Here 是完整的错误消息。请注意，如果我将 m1 和 m2 定义为具有固定大小，则可以正常工作。

我正在使用Eigen3.3.1。它在运行 OSX-10.12 并使用 Apple 的 clang-800.0.42.1 的 Mac 上进行了测试。

Answer 1

这是因为一般的matrix-matrix产品经过了高度优化，采用了激进的手动向量化、流水线、多级缓存等。这部分不支持混合float和double。您可以使用m1.lazyProduct(m2) 绕过这个高度优化的实现，它对应于用于小型固定大小矩阵的实现，但这样做只有缺点：ALU 不支持混合浮点和双精度，因此必须提升浮点值无论如何加倍，你将失去矢量化。最好将浮点数显式转换为双精度：

m1.cast<double>() * m2