Method
__init__
(self, dim: int, n_fft: int, hop_length: int, padding: str = "same")
Source from the content-addressed store, hash-verified
| 481 | """ |
| 482 | |
| 483 | def __init__(self, dim: int, n_fft: int, hop_length: int, padding: str = "same"): |
| 484 | super().__init__() |
| 485 | self.hop_length = hop_length |
| 486 | out_dim = n_fft + 2 |
| 487 | self.out = torch.nn.Linear(dim, out_dim) |
| 488 | self.istft = ISTFT( |
| 489 | n_fft=n_fft, hop_length=hop_length, win_length=n_fft, padding=padding |
| 490 | ) |
| 491 | |
| 492 | def forward(self, x: torch.Tensor, x_len: torch.Tensor) -> torch.Tensor: |
| 493 | """ |
Callers
nothing calls this directly
Tested by
no test coverage detected