a NSicg@sddlZddlmZmZmZmZddlmZmZm Z m Z ddlm Z gdZ d ee eejfe d d d Zee d d dZd!e ee eejfddddZee ddddZe ddddZe ddddZdd ddZdd ddZe d ddZdS)"N)castIterableListUnion) _lazy_init _lazy_call device_countcurrent_device)Tensor) get_rng_stateget_rng_state_all set_rng_stateset_rng_state_all manual_seedmanual_seed_allseedseed_all initial_seedcuda)devicereturncCsZtt|trt|}nt|tr2td|}|j}|durFt}tjj |}| S)aSReturns the random number generator state of the specified GPU as a ByteTensor. Args: device (torch.device or int, optional): The device to return the RNG state of. Default: ``'cuda'`` (i.e., ``torch.device('cuda')``, the current CUDA device). .. warning:: This function eagerly initializes CUDA. rN) r isinstancestrtorchrintindexr rdefault_generators get_state)ridxdefault_generatorr"M/var/www/html/django/DPS/env/lib/python3.9/site-packages/torch/cuda/random.pyr s      r )rcCs&g}ttD]}|t|q|S)zRReturns a list of ByteTensor representing the random number states of all devices.)ranger appendr )resultsir"r"r#r"sr) new_staterrcsT|jtjdttr$tnttr:tdfdd}t|dS)a0Sets the random number generator state of the specified GPU. Args: new_state (torch.ByteTensor): The desired state device (torch.device or int, optional): The device to set the RNG state. Default: ``'cuda'`` (i.e., ``torch.device('cuda')``, the current CUDA device). ) memory_formatrcs6ttjj}|durt}tjj|}|dSN)rrrrr rr set_stater r!rZnew_state_copyr"r#cb9s  zset_rng_state..cbN)clonercontiguous_formatrrrrr)r(rr.r"r-r#r+s    r) new_statesrcCs t|D]\}}t||qdS)zSets the random number generator state of all devices. Args: new_states (Iterable of torch.ByteTensor): The desired state for each deviceN) enumerater)r1r'stater"r"r#rCsr)rrcs$tfdd}t|dddS)aSets the seed for generating random numbers for the current GPU. It's safe to call this function if CUDA is not available; in that case, it is silently ignored. Args: seed (int): The desired seed. .. warning:: If you are working with a multi-GPU model, this function is insufficient to get determinism. To seed all GPUs, use :func:`manual_seed_all`. cs t}tjj|}|dSr*)r rrrrr,rr"r#r.Zs zmanual_seed..cbTr4Nrrrr.r"r4r#rLs  rcs$tfdd}t|dddS)zSets the seed for generating random numbers on all GPUs. It's safe to call this function if CUDA is not available; in that case, it is silently ignored. Args: seed (int): The desired seed. cs*ttD]}tjj|}|q dSr*)r$r rrrr)r'r!r4r"r#r.ls zmanual_seed_all..cbT)rNr5r6r"r4r#rbs rcCsdd}t|dS)atSets the seed for generating random numbers to a random number for the current GPU. It's safe to call this function if CUDA is not available; in that case, it is silently ignored. .. warning:: If you are working with a multi-GPU model, this function will only initialize the seed on one GPU. To initialize all GPUs, use :func:`seed_all`. cSst}tjj|}|dSr*)r rrrrr,r"r"r#r.}s zseed..cbNrr.r"r"r#rts rcCsdd}t|dS)zSets the seed for generating random numbers to a random number on all GPUs. It's safe to call this function if CUDA is not available; in that case, it is silently ignored. cSsLd}d}ttD]4}tjj|}|s<||}d}q||qdS)NrFT)r$r rrrrrr)Z random_seedZseededr'r!r"r"r#r.s zseed_all..cbNr7r8r"r"r#rs rcCs tt}tjj|}|S)zzReturns the current random seed of the current GPU. .. warning:: This function eagerly initializes CUDA. )rr rrrrr,r"r"r#rs r)r)r)rtypingrrrrrrr r r __all__rrrr rrrrrrrrr"r"r"r#s