Function autograd::ops::reduce_mean

pub fn reduce_mean(x: &Tensor, axis: isize, keep_dim: bool) -> Tensor

Take mean along specified axis.