gather function

VARP gather(
  1. VARP input,
  2. VARP indices
)

Gather slices from params 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.

Returns:

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

Implementation

VARP gather(VARP input, VARP indices) => VARP.fromPointer(C.mnn_expr_Gather(input.ptr, indices.ptr));