Source code for espnet2.enh.diffusion.abs_diffusion

from abc import ABC, abstractmethod

import torch


[docs]class AbsDiffusion(torch.nn.Module, ABC):
[docs]    @abstractmethod
    def forward(
        self,
        input: torch.Tensor,
        ilens: torch.Tensor,
    ):
        raise NotImplementedError

[docs]    @abstractmethod
    def enhance(self, input: torch.Tensor):
        raise NotImplementedError