嘉渊 hace 2 años
padre
commit
3039ae315c
Se han modificado 1 ficheros con 14 adiciones y 0 borrados
  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