Rust编写最小自动微分引擎

2025-06-20
Rust编写最小自动微分引擎

这是一个用Rust编写的最小自动微分引擎。该引擎可以训练一个小型多层感知器来学习XOR函数,并能将单个感知器的计算图渲染到graph.html文件中。其核心是Scalar结构体,存储值、梯度和描述生成它的操作的Edge。运算符重载和辅助函数构建一个有向无环图,缓存每个边的局部导数。backward()函数从输出节点开始递归地传播梯度,累积到用Scalar::new_grad创建的叶节点中。计算图可以通过plot::dump_graph可视化。

开发