del lm

wucong.lyb

2023-05-17 f83f3e598538f5f4d1d2c6a6592f976bcc225447

 funasr/layers/abs_normalize.py

New file
@@ -0,0 +1,14 @@
from abc import ABC
from abc import abstractmethod
from typing import Tuple

import torch


class AbsNormalize(torch.nn.Module, ABC):
    @abstractmethod
    def forward(
        self, input: torch.Tensor, input_lengths: torch.Tensor = None
    ) -> Tuple[torch.Tensor, torch.Tensor]:
        # return output, output_lengths
        raise NotImplementedError