    def _invert_exec_key(self, key: str):
        """Invert execution key to get input arguments as integers.

        Parameters
        ----------
        key : str
            Execution key

        Returns
        ----------
            List[int]
        """
        res = []
        for item in re.split(" == | && ", key):
            if item.isnumeric():
                res.append(int(item))
        return res
Invert execution key to get input arguments as integers.

Parameters
----------
key : str
Execution key

Returns
----------
List[int]
python/aitemplate/compiler/ops/softmax/softmax.py