    def _invert_exec_key(self, key):
        """extract tensor shape info from key"""
        return common.gemm_inverse_key_func(key)
extract tensor shape info from key
python/aitemplate/compiler/ops/gemm_epilogue_vistor/bmm_rcr_softmax.py