argMax function

VARP argMax(
  1. VARP x,
  2. int axis
)

Returns the index with the largest value across axes of a tensor.

Args:

  • input: A variable. Must be one of the following types: Halide_Type_Float, Halide_Type_Int
  • axis: A int. must be in the range -rank(input), rank(input)). Describes which axis of the input variable to reduce across. For vectors, use axis = 0.

Returns:

  • A variable of type int.

Implementation

VARP argMax(VARP x, int axis) => VARP.fromPointer(C.mnn_expr_ArgMax(x.ptr, axis));