def test_multi_batch() -> None:
        """Test visual association score computation with multiple batches."""
        embeddings = torch.randn(10, 100, 128)
        reference_embeddings = torch.randn(2, 100, 128)
        assert visual_association_score(embeddings, reference_embeddings).shape == torch.Size([10, 100])
Test visual association score computation with multiple batches.
tests/unit/models/image/winclip/test_utils.py