Function autograd::ops::square [−] [src]

pub fn square(a: &Tensor) -> Tensor

Returns square of the input.

extern crate ndarray;
extern crate autograd as ag;

let ref a = ag::constant(ndarray::arr1(&[2., 3.]));
let ref b = ag::square(a);
assert_eq!(
    b.eval(&[]),
    Ok(ndarray::arr1(&[4., 9.]).into_dyn())
);