paddlespeech.t2s.models.vits.monotonic_align package
Maximum path calculation module.
This code is based on https://github.com/jaywalnut310/vits.
- paddlespeech.t2s.models.vits.monotonic_align.maximum_path(neg_x_ent: Tensor, attn_mask: Tensor) Tensor [source]
Calculate maximum path.
- Args:
neg_x_ent (Tensor): Negative X entropy tensor (B, T_feats, T_text). attn_mask (Tensor): Attention mask (B, T_feats, T_text).
- Returns:
Tensor: Maximum path tensor (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)[source]
Calculate a single maximum path with numba.
- paddlespeech.t2s.models.vits.monotonic_align.maximum_path_numba(paths, values, t_ys, t_xs)[source]
Calculate batch maximum path with numba.