Struct ArgReduceArgs 

pub struct ArgReduceArgs<'a, T: Element, const N: usize, I: IndexOutputElement = i64> {
    pub x: TensorRef<'a, T, N>,
    pub y: TensorMut<'a, I, N>,
}

Args bundle for an arg-reduction launch.

Note the asymmetric dtypes: x is the value dtype T, y is the index dtype I (defaults to i64 — PyTorch convention).

Fields

x: TensorRef<'a, T, N>

Input.

y: TensorMut<'a, I, N>

Output indices — shape matches input with reduce axis = 1. Type parameter I selects u32, i32, or i64 (default).