File size: 493 Bytes
ad16788 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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
|