espnet2/enh/decoder/null_decoder.py

import torch

from espnet2.enh.decoder.abs_decoder import AbsDecoder


class NullDecoder(AbsDecoder):
    """Null decoder, return the same args."""

    def __init__(self):
        super().__init__()

    def forward(self, input: torch.Tensor, ilens: torch.Tensor):
        """Forward. The input should be the waveform already.

        Args:
            input (torch.Tensor): wav [Batch, sample]
            ilens (torch.Tensor): input lengths [Batch]
        """
        return input, ilens