https://github.com/bokutotu/autograd

https://github.com/bokutotu/autograd

Last synced: 12 days ago
JSON representation

• Host: GitHub
• URL: https://github.com/bokutotu/autograd
• Owner: bokutotu
• License: mit
• Created: 2021-05-06T08:38:12.000Z (over 3 years ago)
• Default Branch: main
• Last Pushed: 2022-02-06T05:14:21.000Z (almost 3 years ago)
• Last Synced: 2023-04-25T19:26:05.588Z (over 1 year ago)
• Language: Rust
• Size: 8.79 KB
• Stars: 9
• Watchers: 2
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# Autograd

このクレートはRustで自動微分を実装したクレートです。

簡単に自動微分を実装できます。

# Example

```rust

use autograd::function::*;

use autograd::node::Node;

use autograd::tensor::Tensor;

fn main() {

    // 勾配情報を持つtensorの定義(この場合はスカラー)

    let x = Tensor::new(&[]);

    // y = x * x + xの計算グラフを定義する

    let y = add(&product(&x, &x), &x);

    // 計算グラフの購買情報をリセット

    y.zero_grad();

    

    // 購買情報を伝搬する前にグラフの最も下のノードの勾配をセットする

    y.set_grad();

    

    // 計算グラフの計算を行う

    y.forward();

    // 計算グラフの勾配を計算する

    y. backward();

}

```