torchaudio.functional.rnnt_loss¶
- torchaudio.functional.rnnt_loss(logits: Tensor, targets: Tensor, logit_lengths: Tensor, target_lengths: Tensor, blank: int = -1, clamp: float = -1, reduction: str = 'mean', fused_log_softmax: bool = True)[source]¶
从使用循环神经网络进行序列转导[Graves, 2012]计算 RNN 转导损失。
RNN 转导损失通过定义对所有长度的输出序列的分布，以及通过共同建模输入-输出和输出-输出依赖关系来扩展 CTC 损失。
- 参数:
logits (Tensor) – 尺寸为 (batch, max seq length, max target length + 1, class) 的张量，包含来自连接器的输出
targets (Tensor) – 尺寸为 (batch, max target length) 的张量，包含零填充的目标
logit_lengths (Tensor) – 尺寸为 (batch) 的张量，包含来自编码器的每个序列的长度
target_lengths (Tensor) – 尺寸为 (batch) 的张量，包含每个序列的目标长度
blank (int, optional) – 空白标签 (默认值：
-1)clamp (float, optional) – 梯度钳位 (默认值：
-1)reduction (string, optional) – 指定要应用于输出的缩减：
"none"|"mean"|"sum". (默认值："mean")fused_log_softmax (bool) – 如果在损失之外调用 log_softmax，则设置为 False (默认值：
True)
- 返回值:
应用了缩减选项的损失。如果
reduction为"none"，则尺寸为 (batch)，否则为标量。- 返回类型:
Tensor
