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

yoyolicoris · web-flow · commit 03a6ecc26617 · 2024-11-27T18:40:05.000+08:00
fix: correct tensor broadcasting in avg function
diff --git a/torchcomp/__init__.py b/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,
     )
 

Original file line number	Diff line number	Diff line change
`@@ -89,7 +89,7 @@ def avg(rms: torch.Tensor, avg_coef: Union[torch.Tensor, float]):`
`89`	`89`	`assert torch.all(avg_coef > 0) and torch.all(avg_coef <= 1)`
`90`	`90`
`91`	`91`	`return sample_wise_lpc(`
`92`		`- rms * avg_coef,`
	`92`	`+ rms * avg_coef.unsqueeze(1),`
`93`	`93`	`avg_coef[:, None, None].broadcast_to(rms.shape + (1,)) - 1,`
`94`	`94`	`)`
`95`	`95`