def pop(self, key: str) -> Any:
        r"""Remove key from the ParameterDict and return its parameter.

        Args:
            key (str): key to pop from the ParameterDict
        """
        v = self[key]
        del self[key]
        return v
Remove key from the ParameterDict and return its parameter.

        Args:
            key (str): key to pop from the ParameterDict
        
python/aitemplate/frontend/nn/container.py