o }o™ikã@sddlZdd„ZdS)éNcCsB|jd}t |dd¡ |j¡d|}t |¡ ¡}||}|S)zÞ Function to perform random dropout for embeddings. When we drop embeddings, we zero them out. Args: embeddings (tensor): Input embeddings drop_rate (float): Rate of dropping the embedding. ré)ÚshapeÚtorchÚonesÚtoÚdtypeÚ bernoulliÚcuda)Ú embeddingsÚ drop_rateÚnsamplesÚ zero_flag©rúb/home/ubuntu/.local/lib/python3.10/site-packages/nemo/collections/multimodal/parts/imagen/utils.pyÚrandom_dropouts r)rrrrrrÚs