def test_multi_batch() -> None:
        """Test harmonic aggregation computation with multiple batches."""
        window_scores = torch.randn(2, 4)
        output_size = (3, 3)
        masks = torch.tensor([[0, 1, 3, 4], [1, 2, 4, 5], [3, 4, 6, 7], [4, 5, 7, 8]])
        output = harmonic_aggregation(window_scores, output_size, masks)
        assert output.shape == torch.Size([2, 3, 3])
Test harmonic aggregation computation with multiple batches.
tests/unit/models/image/winclip/test_utils.py