(self, audio_input)
| 529 | return self.data_from_latent(latent) |
| 530 | |
| 531 | def forward(self, audio_input): |
| 532 | x = self.encode(audio_input) |
| 533 | |
| 534 | l_in = x.transpose(1, 2) |
| 535 | l, latent = self.compressor(l_in) |
| 536 | |
| 537 | xhat = self.decode(l) |
| 538 | return xhat, latent |
| 539 | |
| 540 | |
| 541 |