Skip to content

Commit

Permalink
use math block for avg filter
Browse files Browse the repository at this point in the history
  • Loading branch information
yoyolicoris authored Mar 31, 2024
1 parent 1d897cf commit 9ca43be
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -231,13 +231,13 @@ def avg(rms: torch.Tensor, avg_coef: Union[torch.Tensor, float]):

### Equations

$$
```math
\hat{x}_{\rm rms}[n] = \alpha_{\rm avg} x_{\rm rms}[n] + (1 - \alpha_{\rm avg}) \hat{x}_{\rm rms}[n-1]
$$
```

## TODO

- [x] CUDA acceleration in Numba
- [ ] PyTorch CPP extension
- [ ] Native CUDA extension
- [ ] Examples
- [ ] Examples

0 comments on commit 9ca43be

Please sign in to comment.