FunASR/funasr/layers/inversible_interface.py
2023-04-24 15:12:42 +08:00

14 lines
348 B
Python

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