Source code for espnet2.uasr.discriminator.abs_discriminator

from abc import ABC, abstractmethod

import torch


class AbsDiscriminator(torch.nn.Module, ABC):
    @abstractmethod
    def forward(
        self,
        xs_pad: torch.Tensor,
        padding_mask: torch.Tensor,
    ) -> torch.Tensor:
        raise NotImplementedError