Function autograd::ops::ones

pub fn ones<T: ArrayLike>(shape: &T) -> Tensor

Returns ones with given shape

extern crate ndarray;
extern crate autograd as ag;

let a = ag::ones(&[4, 2]);
assert_eq!(a.eval(&mut ag::Context::new()),
                  ndarray::Array2::<f32>::from_elem((4, 2), 1.).into_dyn());