RustAD - Rust 自动微分

RustAD是一个尝试在Rust中实现自动微分的库的早期WIP版本。

为什么我会选择这个而不是其他库？ 目前不会，我会说等到ndarray的支持更加全面时，这个库可能成为最方便的Rust AutoDiff库。

代码很混乱，请小心。

状态

• 前向自动微分
• 反向自动微分
• 数值原语（例如 f32，u32 等）支持*
• 有限的° ndarray 支持*
• 有限的° nalgebra 支持*
• 支持 if，if else 和 else
• 支持 for，while 和 loop

*typeof（例如 decltype）在Rust中尚未实现，使得支持变得更加困难。

°支持限于基本的BLAS-like操作。

应用

自动微分通过2个属性过程宏实现，例如

#[rust_ad::forward_autodiff]
fn multi(x: f32, y: f32) -> f32 {
    let a = x.powi(2i32);
    let b = x * 2f32;
    let c = 2f32 / y;
    let f = a + b + c;
    return f;
}
fn main() {
    let (f, der_x, der_y) = rust_ad::forward!(multi, 3f32, 5f32);
    assert_eq!(f, 15.4f32);
    assert_eq!(der_x, 8f32);
    assert_eq!(der_y, -0.08f32);
}

#[rust_ad::reverse_autodiff]
fn multi(x: f32, y: f32) -> f32 {
    let a = x.powi(2i32);
    let b = x * 2f32;
    let c = 2f32 / y;
    let f = a + b + c;
    return f;
}
fn main() {
    let (f, der_x, der_y) = rust_ad::reverse!(multi, 3f32, 5f32);
    assert_eq!(f, 15.4f32);
    assert_eq!(der_x, 8f32);
    assert_eq!(der_y, -0.08f32);
}