def test_single_batch() -> None:
        """Test cosine similarity with single batch inputs."""
        input1 = torch.randn(1, 100, 128)
        input2 = torch.randn(1, 200, 128)
        assert cosine_similarity(input1, input2).shape == torch.Size([1, 100, 200])
Test cosine similarity with single batch inputs.
tests/unit/models/image/winclip/test_utils.py