funasr/layers/inversible_interface.py
New file @@ -0,0 +1,14 @@ from abc import ABC from abc import abstractmethod from typing import Tuple import torch class InversibleInterface(ABC): @abstractmethod def inverse( self, input: torch.Tensor, input_lengths: torch.Tensor = None ) -> Tuple[torch.Tensor, torch.Tensor]: # return output, output_lengths raise NotImplementedError