Merge pull request #7 from DiffAPF:fix-avg_coef-shape

fix: correct tensor broadcasting in avg function

torchcomp/__init__.py

@@ -89,7 +89,7 @@ def avg(rms: torch.Tensor, avg_coef: Union[torch.Tensor, float]):
     assert torch.all(avg_coef > 0) and torch.all(avg_coef <= 1)
 
     return sample_wise_lpc(
-        rms * avg_coef,
+        rms * avg_coef.unsqueeze(1),
         avg_coef[:, None, None].broadcast_to(rms.shape + (1,)) - 1,
     )