torchaudio.functional.forced_align¶
- torchaudio.functional.forced_align(log_probs: Tensor, targets: Tensor, input_lengths: Optional[Tensor] = None, target_lengths: Optional[Tensor] = None, blank: int = 0) Tuple[Tensor, Tensor] [source]¶
将 CTC 标签序列与发射对齐。
- 参数:
log_probs (Tensor) – CTC 发射输出的对数概率。形状为 (B, T, C) 的张量。其中 B 是批次大小,T 是输入长度,C 是字母表中字符的数量,包括空白。
targets (Tensor) – 目标序列。形状为 (B, L) 的张量,其中 L 是目标长度。
input_lengths (Tensor or None, optional) – 输入的长度(最大值必须小于或等于 T)。形状为 (B,) 的一维张量。
target_lengths (Tensor or None, optional) – 目标的长度。形状为 (B,) 的一维张量。
blank_id (int, optional) – CTC 发射中空白符号的索引。(默认:0)
- 返回值:
Tensor: 使用强制对齐计算的对齐路径中每个时间步的标签。
Tensor: 每个时间步的标签的对数概率得分。
- 返回类型:
Tuple(Tensor, Tensor)
注意
log_probs 的序列长度必须满足
\[L_{\text{log\_probs}} \ge L_{\text{label}} + N_{\text{repeat}} \]其中 \(N_{\text{repeat}}\) 是连续重复的标记数量。例如,在字符串 “aabbc” 中,重复次数为 2。
注意
当前版本仅支持
batch_size==1
。- 使用
forced_align
的教程