def test_2d_3d() -> None:
        """Test cosine similarity with 2D and 3D input."""
        input1 = torch.randn(100, 128)
        input2 = torch.randn(1, 200, 128)
        assert cosine_similarity(input1, input2).shape == torch.Size([100, 200])
Test cosine similarity with 2D and 3D input.
tests/unit/models/image/winclip/test_utils.py