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