嘉渊 2 anni fa
parent
commit
3039ae315c
1 ha cambiato i file con 14 aggiunte e 0 eliminazioni
  1. 14 0
      funasr/layers/inversible_interface.py

+ 14 - 0
funasr/layers/inversible_interface.py

@@ -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