autograd 0.8.0

Tensors and differentiable operations in Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

extern crate autograd as ag;

struct MultiOutputOp;

impl ag::op::Op for MultiOutputOp {
    fn name(&self) -> &str {
        "MultiOutputOp"
    }

    fn compute(&self, _: ag::runtime::OpComputeContext) -> ag::op::ComputeResult {
        let a = ag::ndarray_ext::zeros(&[2, 3]);
        let b = ag::ndarray_ext::zeros(&[1, 3]);
        vec![Ok(a), Ok(b)]
    }

    fn grad(&self, _: &ag::Tensor, _: &[&ag::Tensor], _: &ag::Tensor) -> Vec<Option<ag::Tensor>> {
        vec![None; 2]
    }
}

#[test]
fn test_nth_tensor() {
    let ref a = ag::Tensor::builder().build(MultiOutputOp);
    let ref b = ag::nth_tensor(a, 1);
    let ref c = ag::exp(b);
    ag::eval(&[c], &[]);
}