Source code for espnet2.asr.postencoder.abs_postencoder

from abc import ABC, abstractmethod
from typing import Tuple

import torch


[docs]class AbsPostEncoder(torch.nn.Module, ABC):
[docs] @abstractmethod def output_size(self) -> int: raise NotImplementedError
[docs] @abstractmethod def forward( self, input: torch.Tensor, input_lengths: torch.Tensor ) -> Tuple[torch.Tensor, torch.Tensor]: raise NotImplementedError