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(&[]), Ok(ndarray::Array2::<f32>::from_elem((4, 2), 1.).into_dyn()));