o ¡¿¯iáã@s(ddlZddlZGdd„dejjƒZdS)éNcs,eZdZdZd‡fdd„ Zd dd„Z‡ZS) ÚGaussianUpsamplingz]Gaussian upsampling with fixed temperature as in: https://arxiv.org/abs/2010.04301 çš™™™™™¹?cstƒ ¡||_dS)N)ÚsuperÚ__init__Údelta)Úselfr©Ú __class__©úY/home/ubuntu/.local/lib/python3.10/site-packages/espnet2/gan_tts/jets/length_regulator.pyrs  zGaussianUpsampling.__init__Nc Cs| d¡}|j}| ¡dkrt d¡d||jdd d¡<|dur)| ¡ ¡}n| d¡}t d|¡  d¡  |d¡  |¡  ¡}|durJ||  ¡}|j dd|d} d|j|  d¡|   d¡d} |dur{|  |  d¡  d|d¡t dƒ ¡} tj| dd} t | |¡}|S) a’Upsample hidden states according to durations. Args: hs (Tensor): Batched hidden state to be expanded (B, T_text, adim). ds (Tensor): Batched token duration (B, T_text). h_masks (Tensor): Mask tensor (B, T_feats). d_masks (Tensor): Mask tensor (B, T_text). Returns: Tensor: Expanded hidden state (B, T_feat, adim). rzLpredicted durations includes all 0 sequences. fill the first element with 1.é)ÚdimNéÿÿÿÿéÚinf)ÚsizeÚdeviceÚsumÚloggingÚwarningÚeqÚintÚtorchÚarangeÚ unsqueezeÚrepeatÚtoÚfloatÚcumsumrÚ masked_fillÚsoftmaxÚmatmul) rÚhsÚdsÚh_masksÚd_masksÚBrÚT_featsÚtÚcÚenergyÚp_attnr r r Úforwards,  ÿ $ "ÿ zGaussianUpsampling.forward)r)NN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rr,Ú __classcell__r r rr r sr)rrÚnnÚModulerr r r r Ús