Function _subtensor

imperative/python/megengine/module/vision.py:274–281

(src, axis, begin, end)

        c_3 = c(3, dtype="int32")

        def _subtensor(src, axis, begin, end):
            items = ((axis, (begin is not None), (end is not None), False, False),)
            args = ()
            if begin is not None:
                args += (begin,)
            if end is not None:
                args += (end,)
            return f(builtin.Subtensor(items=items), src, *args)

        def _kernel_init(x):
            k_1 = _subtensor(x, 0, None, c_1)