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());