mirror of https://github.com/coqui-ai/TTS.git
20 lines
410 B
Python
20 lines
410 B
Python
import torch
|
|
import numpy as np
|
|
|
|
|
|
def alignment_diagonal_score(alignments):
|
|
"""
|
|
Compute how diagonal alignment predictions are. It is useful
|
|
to measure the alignment consistency of a model
|
|
Args:
|
|
alignments (torch.Tensor): batch of alignments.
|
|
Shape:
|
|
alignments : batch x decoder_steps x encoder_steps
|
|
"""
|
|
return alignments.max(dim=1)[0].mean(dim=1).mean(dim=0)
|
|
|
|
|
|
|
|
|
|
|