def get_empty_buffer(self, *shape) -> Union["Tensor", OrtValue]:
        """Get an empty buffer with the given shape.

        :param shape: Shape of the buffer.
        :return: Buffer with the given shape with zeros. If backend is "torch", returns a torch.Tensor.
            If backend is "ort", returns an OrtValue.
        """
        if self.backend == "torch":
            import torch

            return torch.zeros(shape, dtype=self.torch_dtype, device=self.torch_device)
        return self.get_empty_ortvalue(*shape)
Get an empty buffer with the given shape.

        :param shape: Shape of the buffer.
        :return: Buffer with the given shape with zeros. If backend is "torch", returns a torch.Tensor.
            If backend is "ort", returns an OrtValue.
        
examples/utils/kv_cache_utils.py