Function max

pub fn max<'a>(
    lhs: &'a Tensor<'_>,
    dim: isize,
) -> Result<Tensor<'a>, TensorError>