Function autograd::ops::normalize

pub fn normalize<T: ArrayLike>(x: &Tensor, axes: &T) -> Tensor

Normalizes input tensor with its mean and variance along specified axis.

extern crate ndarray;
extern crate autograd as ag;

let ref x = ag::standard_normal(&[3, 4]);
let ref y1 = ag::normalize(x, &[0]);
let ref y2 = ag::normalize(x, &[0]);

let evaluated = ag::eval(&[y1, y2], &mut ag::Context::new());
assert_eq!(&[3, 4], evaluated[0].shape());
assert_eq!(&[3, 4], evaluated[1].shape());