FunASR/funasr/layers/inversible_interface.py
2022-11-26 21:56:51 +08:00

15 lines
349 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