o
    ¡¿¯ií  ã                   @   s(   d dl Z d dlmZ G dd„ deƒZdS )é    N)Ú
AbsDecoderc                       s6   e Zd ZdZ‡ fdd„Zdejdejfdd„Z‡  ZS )ÚNullDecoderz#Null decoder, return the same args.c                    s   t ƒ  ¡  d S )N)ÚsuperÚ__init__)Úself©Ú	__class__© úT/home/ubuntu/.local/lib/python3.10/site-packages/espnet2/enh/decoder/null_decoder.pyr   	   s   zNullDecoder.__init__ÚinputÚilensc                 C   s   ||fS )z¸Forward. The input should be the waveform already.

        Args:
            input (torch.Tensor): wav [Batch, sample]
            ilens (torch.Tensor): input lengths [Batch]
        r	   )r   r   r   r	   r	   r
   Úforward   s   zNullDecoder.forward)	Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   ÚtorchÚTensorr   Ú__classcell__r	   r	   r   r
   r      s    r   )r   Úespnet2.enh.decoder.abs_decoderr   r   r	   r	   r	   r
   Ú<module>   s    