1
2
3
4
5
6
7
8
9
10
11
12
13
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
|
|