RNNTBundle.FeatureExtractor

class torchaudio.pipelines.RNNTBundle.FeatureExtractor

Interface of the feature extraction part of RNN-T pipeline

__call__

abstract FeatureExtractor.__call__(input: Tensor) → Tuple[Tensor, Tensor]

Generates features and length output from the given input tensor.

Parameters:

input (torch.Tensor) – input tensor.

Returns:

torch.Tensor:

Features, with shape (length, *).

torch.Tensor:

Length, with shape (1,).

Return type:

(torch.Tensor, torch.Tensor)