o –h–ã@s@ddlZddlZddlZddlmZejjejjgZdd„Z dS)éN)Ú WordDropoutcCs¦tdƒ}t dd¡}||ƒ}t ||¡rJ‚d}t|jdƒD]$}t ||||¡s6t ||¡dks6J‚t ||¡dkrC|d7}q|dkrO||jdksQJ‚dS)a Test that word_dropout is randomly dropping out the entire final dimension of a tensor Doing 600 small rows should be super fast, but it leaves us with something like a 1 in 10^180 chance of the test failing. Not very common, in other words gà?iXérgéN)rÚtorchÚrandnÚallcloseÚrangeÚshapeÚsum)ÚwdÚbatchÚdroppedÚ num_zerosÚi©rú[/var/www/html/env_mimamsha/lib/python3.10/site-packages/stanza/tests/common/test_dropout.pyÚtest_word_dropout s *€r) ÚpytestrÚstanzaÚstanza.models.common.dropoutrÚmarkÚtravisÚpipelineÚ pytestmarkrrrrrÚs