Source code for espnet2.spk.pooling.abs_pooling

from abc import ABC, abstractmethod

import torch


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

    @abstractmethod
    def output_size(self) -> int:
        raise NotImplementedError