Function PyMNNVar_subscript

pymnn/src/expr.h:531–583

    return isbool;
}
static PyObject* PyMNNVar_subscript(PyObject* x, PyObject* slice) {
    // gather: 1. 0-1 gather; 2. idx gather;
    if (isIdx(slice)) {
        auto val = toVar(x);
        auto idx = toVar(slice);
        if (nullptr == val->getInfo()) {
            PyMNN_ERROR("Can't support subscript for tensor without shape\n");
            Py_RETURN_NONE;
        }
        if (val->getInfo()->size > 1 && isBoolIdx(idx, val->getInfo()->size)) {
            // 0-1 gather -> idx gather
            idx = Express::_Where(idx);
            val = Express::_GatherND(val, idx);
            val = Express::_Reshape(val, {-1});
            return toPyObj(val);
        }
        auto r = Express::_Gather(val, idx);
        r->readMap<void>();
        return toPyObj(r);
    }

    std::vector<int> begin, end, strides;
    int new_axis_mask = 0, shrink_axis_mask = 0, begin_mask = 0, end_mask = 0, ellipsis_mask = 0;
    dealSlice(slice, begin, end, strides, new_axis_mask, shrink_axis_mask, begin_mask, end_mask, ellipsis_mask);
    int size_ = static_cast<int>(begin.size());
    auto begin_ = Express::_Const(begin.data(), {size_}, NHWC, halide_type_of<int>());
    auto end_ = Express::_Const(end.data(), {size_}, NHWC, halide_type_of<int>());
    auto strides_ = Express::_Const(strides.data(), {size_}, NHWC, halide_type_of<int>());
    auto res = Express::_StridedSlice(toVar(x), begin_, end_, strides_, begin_mask, end_mask,
                                      ellipsis_mask, new_axis_mask, shrink_axis_mask);
    auto info = res->getInfo();
    if (!info) {
        MNN_ERROR("subscript: unable to get variable info");
        Py_RETURN_NONE;
    }
    // to scalar
    if (info->dim.empty()) {
        auto dtype = info->type;
        if (dtype == halide_type_of<float>()) {
            return toPyObj(res->readMap<float>()[0]);
        }
        if (dtype == halide_type_of<int>()) {
            return toPyObj(res->readMap<int>()[0]);
        }
        if (dtype == halide_type_of<uint8_t>()) {
            return toPyObj(res->readMap<uint8_t>()[0]);
        }
        if (dtype == halide_type_of<double>()) {
            return toPyObj((float)res->readMap<double>()[0]);
        }
    }
    return toPyObj(res);
}

static int PyMNNVar_ass_subscript(PyObject* x, PyObject* slice, PyObject* y) {
    if (!isVar(x) || !isVar(y)) {