gatherND function

VARP gatherND(
  1. VARP params,
  2. VARP indices
)

Gather slices from params into a variable with shape specified by indices.

Args:

  • params: A variable. The variables from which to gather values.
  • indices: A variable. Must be one of the following types: Halide_Type_Int.

Returns:

  • A variable. Has the same type as params.

Implementation

VARP gatherND(VARP params, VARP indices) => VARP.fromPointer(C.mnn_expr_GatherND(params.ptr, indices.ptr));