From 3039ae315ce23575e7f5b10904ac23df20499e56 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=98=89=E6=B8=8A?= Date: Mon, 24 Apr 2023 15:12:42 +0800 Subject: [PATCH] update --- funasr/layers/inversible_interface.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) create mode 100644 funasr/layers/inversible_interface.py diff --git a/funasr/layers/inversible_interface.py b/funasr/layers/inversible_interface.py new file mode 100644 index 000000000..657ec6800 --- /dev/null +++ b/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 \ No newline at end of file