Apollo-PyHARP-Rebuild

Sleeping

Apollo-PyHARP-Rebuild / look2hear /utils /inversible_interface.py

initial fr

e617857 9 months ago

333 Bytes

from abc import ABC, 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

	from abc import ABC, 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