Function l2::argmax

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