Function _Sort

express/NeuralNetWorkOp.cpp:1832–1845

}

VARP _Sort(VARP x, int axis, bool arg, bool descend) {
    std::unique_ptr<OpT> op(new OpT);
    op->type = OpType_TopKV2;
    op->main.type = OpParameter_TopKV2;
    auto topk = new TopKV2T;
    topk->largest = descend;
    op->main.value = topk;
    auto posAxis = _Mod(_Scalar(axis), _Rank(x));
    auto K = _Slice(_Shape(x), _Unsqueeze(posAxis, {0}), _Unsqueeze(_Scalar<int32_t>(1), {0}));

    std::vector<VARP> inputs {x, K, _Scalar(axis)};
    auto expr = Expr::create(op.get(), inputs, 2);
    return Variable::create(expr, arg);
}

VARP _Raster(const std::vector<VARP>& vars, const std::vector<int>& region, const std::vector<int>& shape) {
    auto expr = Utils::makeRaster(vars, region, shape, halide_type_of<float>(), MNN_DATA_FORMAT_UNKNOWN);