paddlespeech.t2s.models.vits.monotonic_align 包
最大路径计算模块。
此代码基于 https://github.com/jaywalnut310/vits。
- paddlespeech.t2s.models.vits.monotonic_align.maximum_path(neg_x_ent: Tensor, attn_mask: Tensor) Tensor[来源]
计算最大路径。
- Args:
neg_x_ent (Tensor)：负X熵张量 (B, T_feats, T_text)。
attn_mask (Tensor)：注意力掩码 (B, T_feats, T_text)。- Returns:
张量：最大路径张量 (B, T_feats, T_text)。
- paddlespeech.t2s.models.vits.monotonic_align.maximum_path_each_numba(path, value, t_y, t_x, max_neg_val=-inf)[来源]
使用numba计算单个最大路径。
- paddlespeech.t2s.models.vits.monotonic_align.maximum_path_numba(paths, values, t_ys, t_xs)[来源]
使用numba计算批量最大路径。