Function _empty_like_tensor

KVCOMM/llm/kvcomm_engine.py:179–182

(tensor: Optional[torch.Tensor])

def _empty_like_tensor(tensor: Optional[torch.Tensor]) -> Optional[torch.Tensor]:
    if tensor is None or isinstance(tensor, list):
        return tensor
    return tensor[..., :0, :].clone()


def _ensure_same_layout(cache: DynamicCache, other: DynamicCache) -> None: