o ¡¿¯i&ã@s$dZddlZGdd„dejjƒZdS)zYVITS-related loss modules. This code is based on https://github.com/jaywalnut310/vits. éNc @s>eZdZdZdejdejdejdejdejdejf dd „Zd S) ÚKLDivergenceLosszKL divergence loss.Úz_pÚlogs_qÚm_pÚlogs_pÚz_maskÚreturncCsv| ¡}| ¡}| ¡}| ¡}| ¡}||d}|d||dt d|¡7}t ||¡}|t |¡}|S)aäCalculate KL divergence loss. Args: z_p (Tensor): Flow hidden representation (B, H, T_feats). logs_q (Tensor): Posterior encoder projected scale (B, H, T_feats). m_p (Tensor): Expanded text encoder projected mean (B, H, T_feats). logs_p (Tensor): Expanded text encoder projected scale (B, H, T_feats). z_mask (Tensor): Mask tensor (B, 1, T_feats). Returns: Tensor: KL divergence loss. gà?égÀ)ÚfloatÚtorchÚexpÚsum)ÚselfrrrrrÚklÚloss©rúM/home/ubuntu/.local/lib/python3.10/site-packages/espnet2/gan_tts/vits/loss.pyÚforwards "zKLDivergenceLoss.forwardN)Ú__name__Ú __module__Ú __qualname__Ú__doc__r ÚTensorrrrrrr sþýüûúùr)rr ÚnnÚModulerrrrrÚs