(self, encoded_frame: EncodedFrame)
| 178 | return _linear_overlap_add(frames, self.segment_stride or 1) |
| 179 | |
| 180 | def _decode_frame(self, encoded_frame: EncodedFrame) -> torch.Tensor: |
| 181 | codes, scale = encoded_frame |
| 182 | codes = codes.transpose(0, 1) |
| 183 | emb = self.quantizer.decode(codes) |
| 184 | out = self.decoder(emb) |
| 185 | if scale is not None: |
| 186 | out = out * scale.view(-1, 1, 1) |
| 187 | return out |
| 188 | |
| 189 | def forward(self, x: torch.Tensor) -> torch.Tensor: |
| 190 | frames = self.encode(x) |