o ¡¿¯i#ã@sdZdd„ZdS)z'Attention masking module for Masked LM.cCsD||k d¡}| d¡}| dd¡ dd|¡}| d|d¡|@}|S)a Create attention mask to avoid attending on padding tokens. :param torch.Tensor ys_pad: batch of padded target sequences (B, Lmax) :param int ignore_id: index of padding :param torch.dtype dtype: result dtype :rtype: torch.Tensor (B, Lmax, Lmax) éþÿÿÿéÿÿÿÿéé)Ú unsqueezeÚsizeÚ transposeÚrepeat)Ú ys_in_padÚ ignore_idÚys_maskÚymaxÚ ys_mask_tmp©rú\/home/ubuntu/.local/lib/python3.10/site-packages/espnet/nets/pytorch_backend/maskctc/mask.pyÚ square_mask s  rN)Ú__doc__rrrrrÚs