o j@sdddZdS)FcCs8|jddd}|rd||dk<|jddjddS)aU Compute how diagonal alignment predictions are. It is useful to measure the alignment consistency of a model Args: alignments (torch.Tensor): batch of alignments. binary (bool): if True, ignore scores and consider attention as a binary mask. Shape: - alignments : :math:`[B, T_de, T_en]` )dim)maxmeanitem) alignmentsbinarymaxsr I/home/kuhnn/.local/lib/python3.10/site-packages/TTS/tts/utils/measures.pyalignment_diagonal_scores  r N)F)r r r r r s