表达式模板通过延迟计算和消除临时对象优化数值运算,以Vector加法为例,利用模板将表达式构造成计算树,在赋值时才执行,实现零成本抽象、减少内存分配与自动循环融合,提升性能。
在C++中,表达式模板(Expression Templates)是一种用于编译期优化数值计算的技术,尤其适用于向量、矩阵等数学对象的运算。它通过延迟计算和消除临时对象,显著提升性能。下面以一个简单的向量加法为例,说明如何实现一个基础的表达式模板。
基本向量类设计
首先定义一个容器类
Vector
,用于存储数值数据:
表达式模板的核心思想
当我们写
a = b + c + d
时,传统方式会生成多个临时对象。表达式模板通过模板参数将整个表达式构造成一个“计算树”,在赋值时才真正执行计算。
定义一个基类表示任意表达式:
立即学习
“
C++免费学习笔记(深入)
”;
然后定义加法表达式模板:
C知道
CSDN推出的一款AI技术问答工具
下载
重载操作符与类型推导
为支持表达式组合,需重载
+
操作符:
最后,在赋值操作中触发实际计算:
这样,表达式
v1 = v2 + v3 + v4
不会创建中间临时对象,所有加法在循环中一次性完成,实现了“融合循环”优化。
小结:关键优势
零成本抽象
:模板在编译期展开,运行时无额外开销。
减少内存分配
:避免中间结果的临时存储。
自动循环融合
:多个操作合并为单次遍历。
基本上就这些。表达式模板是C++编译期技巧的经典应用,广泛用于Eigen、Blaze等高性能数值库中。理解其原理有助于写出更高效的数学计算代码。不复杂但容易忽略细节。
template
class Vector {
public:
explicit Vector(size_t size) : data_(size) {}
T& operator[](size_t i) { return data_[i]; }
const T& operator[](size_t i) const { return data_[i]; }
size_t size() const { return data_.size(); }
private:
std::vector data_;
};
template
struct ExprBase {
const Expr& self() const { return static_cast(*this); }
};
template
class AddExpr : public ExprBase> {
public:
AddExpr(const LHS& lhs, const RHS& rhs) : lhs_(lhs), rhs_(rhs) {}
auto operator[](size_t i) const { return lhs_[i] + rhs_[i]; }
size_t size() const { return lhs_.size(); } // 假设大小一致
private:
const LHS& lhs_;
const RHS& rhs_;
};
template
AddExpr operator+(const ExprBase& lhs, const ExprBase>& rhs) {
return AddExpr(lhs.self(), rhs.self());
}
// 使 Vector 成为表达式的一部分
template
class Vector : public ExprBase> {
// ... 同上
};
template
Vector& operator=(Vector& vec, const ExprBase& expr) {
const Expr& e = expr.self();
for (size_t i = 0; i < vec.size(); ++i) {
vec[i] = e[i];
}
return vec;
}
