chela 0.0.2

High-performance Machine Learning, Auto-Differentiation and Tensor Algebra crate for Rust
Documentation
1
2
3
4
5
6
7
8
9
10
11
use chela::*;

fn main() {
    let _grad = {
        let mut a = Tensor::from([1.0f32, 2.0, 3.0]);
        a.set_requires_grad(true);
        a.backward();

        a.gradient().unwrap()
    };
}