Source code for espnet2.enh.decoder.abs_decoder

from abc import ABC, abstractmethod
from typing import Tuple

import torch

[docs]class AbsDecoder(torch.nn.Module, ABC):
[docs] @abstractmethod def forward( self, input: torch.Tensor, ilens: torch.Tensor, fs: int = None, ) -> Tuple[torch.Tensor, torch.Tensor]: raise NotImplementedError
[docs] def forward_streaming(self, input_frame: torch.Tensor): raise NotImplementedError
[docs] def streaming_merge(self, chunks: torch.Tensor, ilens: torch.tensor = None): """Stream merge. It merges the frame-level processed audio chunks in the streaming *simulation*. It is noted that, in real applications, the processed audio should be sent to the output channel frame by frame. You may refer to this function to manage your streaming output buffer. Args: chunks: List [(B, frame_size),] ilens: [B] Returns: merge_audio: [B, T] """ raise NotImplementedError