Minimal auto-differentiation engine in Rust (for educational purposes)
a year ago
- #automatic-differentiation
- #Rust
- #machine-learning
- 用Rust编写的最小化自动微分引擎。
- 演示训练微型多层感知机学习XOR函数。
- 将单个感知机的计算图渲染输出至graph.html文件。
- 使用Scalar对象实现自动微分的示例用法。
- Scalar类型存储值、梯度和运算细节。
- 运算符重载和辅助方法构建有向无环图。
- backward()方法沿计算图递归传播梯度。
- 通过plot::dump_graph实现D3.js可视化图谱导出。