    def _gen_exec_key(self, shape: List[int]):
        """rending the shape info"""
        elem_cnt = np.prod(shape)
        instance_size = shape[-1]
        instance_num = elem_cnt // instance_size
        return self.exec_key_template.render(
            x_dim0=instance_size, x_dim1=instance_num
        ).replace("\n", "")
rending the shape info
python/aitemplate/compiler/ops/tensor/argmax.py