Function autograd::ops::sub [] [src]

pub fn sub(a: &Tensor, b: &Tensor) -> Tensor

Subtraction.

- operator can be used instead.

extern crate ndarray;
extern crate autograd as ag;

let ref a = ag::ones(&[2]);
let ref b = ag::ones(&[2]);

let mut ctx = ag::Context::new();
let ref z: ag::Tensor = a - b;
assert_eq!(z.eval(&mut ctx), ndarray::arr1(&[0., 0.]).into_dyn());