o €o™i"ã @s@ddlmZddlZddlmZd dedededefd d „ZdS) é)ÚAnyN)ÚTensorTÚnÚ ensure_permÚkwargsÚreturnc Kshtj|fi|¤Ž}|r2t t |tj||jd¡¡r2tj|fi|¤Ž}t t |tj||jd¡¡s|S)zL`randomperm` with the ability to ensure the different arrangement generated.)Údevice)ÚtorchÚrandpermÚallÚeqÚaranger)rrrÚperm©rú^/home/ubuntu/.local/lib/python3.10/site-packages/kornia/augmentation/random_generator/utils.pyr s ÿr )T)Útypingrr Ú kornia.corerÚintÚboolr rrrrÚs