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,
) -> Tuple[torch.Tensor, torch.Tensor]:
raise NotImplementedError