Source code for espnet2.enh.extractor.abs_extractor
from abc import ABC, abstractmethod
from collections import OrderedDict
from typing import Dict, Optional, Tuple
import torch
[docs]class AbsExtractor(torch.nn.Module, ABC):
[docs] @abstractmethod
def forward(
self,
input: torch.Tensor,
ilens: torch.Tensor,
input_aux: torch.Tensor,
ilens_aux: torch.Tensor,
suffix_tag: str = "",
additional: Optional[Dict] = None,
) -> Tuple[Tuple[torch.Tensor], torch.Tensor, OrderedDict]:
raise NotImplementedError