Function autograd::ops::mul [] [src]

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

Multiplication.

* operator can be used instead.

extern crate ndarray;
extern crate autograd as ag;

let mut ctx = ag::Context::new();

let ref a = ag::ones(&[2]);
let ref b = ag::ones(&[2]);
let ref z: ag::Tensor = a * b;
assert_eq!(z.eval(&mut ctx), ndarray::arr1(&[1., 1.]).into_dyn());