For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Mojo function

reduce_argmax

def reduce_argmax[dtype: DType, InputFn: def[width: Int, alignment: Int](Coord[**?]) -> SIMD[dtype, width] & RegisterPassable & ImplicitlyCopyable, OutputFn: def[width: SIMDLength](Coord[**?], SIMD[.int64, width]) -> None & RegisterPassable & ImplicitlyCopyable, /, target: StaticString, *, reduce_dim: Int, name: StaticString = "reduce_argmax"](input_fn: InputFn, output_fn: OutputFn, input_shape: Coord[*input_shape.element_types], context: Optional[DeviceContextBase] = None)