gatherV2 function

VARP gatherV2(
  1. VARP input,
  2. VARP indices,
  3. VARP axis
)

Gather slices from params axis according to indices.

Arguments:

  • params: The variable from which to gather values.
  • indices: Index variable. Must be Halide_Type_Int in range 0, ndims(params)-1.
  • axis: A int, the axis in params to gather indices from. Supports negative indexes. If set to 0, it's same as _Gather. Currently only 0 is supported.

Returns:

  • Output: Values from params gathered from indices given by indices.

Implementation

VARP gatherV2(VARP input, VARP indices, VARP axis) =>
    VARP.fromPointer(C.mnn_expr_GatherV2(input.ptr, indices.ptr, axis.ptr));