(
self, n_fft: int, hop_length: int, win_length: int, padding: str = "same"
)
| 336 | """ |
| 337 | |
| 338 | def __init__( |
| 339 | self, n_fft: int, hop_length: int, win_length: int, padding: str = "same" |
| 340 | ): |
| 341 | super().__init__() |
| 342 | assert padding in ["center", "same"], "Padding must be 'center' or 'same'." |
| 343 | self.padding = padding |
| 344 | self.n_fft = n_fft |
| 345 | self.hop_length = hop_length |
| 346 | self.win_length = win_length |
| 347 | window = torch.hann_window(win_length) |
| 348 | self.register_buffer("window", window) |
| 349 | |
| 350 | def forward(self, spec: torch.Tensor) -> torch.Tensor: |
| 351 | """ |