Function autograd::ops::reduce_max

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

Take max along specified axis.