Function q_argmax 

pub fn q_argmax(
    tensor: HostQTensor,
    dim: usize,
    out_dtype: IntDType,
) -> HostTensor