This commit is contained in:
嘉渊 2023-04-24 15:12:42 +08:00
parent 0149983c23
commit 3039ae315c

View 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