Function autograd::ops::sign

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

Returns -1 if x < 0, 0 if x==0, 1 if x > 0, element-wise.

extern crate ndarray;
extern crate autograd as ag;

let mut ctx = ag::Context::new();
let ref a = ag::constant(ndarray::arr1(&[-5., 4.5, 0.]), &mut ctx);
let ref b = ag::sign(a);
assert_eq!(
    b.eval(&mut ctx).as_slice().unwrap(),
    &[-1., 1., 0.]
);