Function _GatherND

express/NeuralNetWorkOp.cpp:1004–1008

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. */

A variable. Has the same type as params.
*/
VARP _GatherND(VARP params, VARP indices) {
    std::unique_ptr<OpT> op(new OpT);
    op->type       = OpType_GatherND;
    return (Variable::create(Expr::create(std::move(op), {params, indices})));
}

VARP _GatherElements(VARP params, VARP indices) {
    std::unique_ptr<OpT> op(new OpT);