ZeNu Autograd

ZeNu Autograd是一个为Rust编写的自动微分库。它通过自动计算数学表达式的梯度为构建和训练神经网络提供基础。

特性

• 使用Variables定义和操作数学表达式
• 通过逆模式自动微分自动计算梯度
• 支持各种数学运算和函数
• 与ZeNu深度学习库集成

入门

要在Rust项目中使用ZeNu Autograd，请将以下内容添加到您的Cargo.toml文件中

[dependencies]
zenu-autograd = "0.1.0"

以下是一个使用ZeNu Autograd的简单示例

use zenu_autograd::{Variable, creator::from_vec::from_vec};

fn main() {
    let x = from_vec(vec![1., 2., 3., 4., 5., 6.], [3, 2]);
    let y = from_vec(vec![7., 8., 9., 10., 11., 12.], [3, 2]);
    let z = x.clone() * y.clone() + y.clone();

    z.backward();

    let x_grad = x.get_grad().unwrap();
    let y_grad = y.get_grad().unwrap();

    // Perform further computations with the gradients
}

有关更多详细信息和示例，请参阅文档。

许可证

ZeNu Autograd在MIT许可证下授权。