Function autograd::ops::ones

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

Returns ones with given shape.

extern crate ndarray;
extern crate autograd as ag;

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