Function autograd::ops::range

pub fn range<T: ArrayLike>(start: &T, end: &T, step: &T) -> Tensor

Returns a range

Unlike range, inputs are symbolic tensors.

extern crate ndarray;
extern crate autograd as ag;

let ref start = ag::scalar(0.);
let ref end = ag::scalar(5.);
let ref step = ag::scalar(1.);
let ref z = ag::range(start, end, step);

assert_eq!(z.eval(&mut ag::Context::new()), ndarray::Array1::range(0., 5., 1.).into_dyn());