Function max

pub fn max<T: ReduceArgs>(x: &Tensor, args: T) -> Tensor