(n, scales=0, dtype=None, device=None)
| 25 | |
| 26 | @lru_cache |
| 27 | def freq_weight_1d(n, scales=0, dtype=None, device=None): |
| 28 | ramp = torch.linspace(0.5 / n, 0.5, n, dtype=dtype, device=device) |
| 29 | weights = -torch.log2(ramp) |
| 30 | if scales >= 1: |
| 31 | weights = torch.clamp_max(weights, scales) |
| 32 | return weights |
| 33 | |
| 34 | |
| 35 | @lru_cache |