IЦi0SSKJrJrJr SSKrSSKJr SSKJrJrJ r J r /SQr SS\\ \ \R4S\4S jjrS\\4S jrSS \S\\ \ \R4SS4S jjrS \\SS4SjrS\ SS4SjrS\ SS4SjrSSjrSSjrS\ 4Sjrg))IterableListUnionN)Tensor) _lazy_call _lazy_initcurrent_device device_count) get_rng_stateget_rng_state_all set_rng_stateset_rng_state_all manual_seedmanual_seed_allseedseed_all initial_seeddevicereturncR[5 [U[5(a[R"U5nO,[U[ 5(a[R"SU5nUR nUc [5n[RRUnUR5$)a:Return 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. cuda) r isinstancestrtorchrintindexr rdefault_generators get_state)ridxdefault_generators P/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/torch/cuda/random.pyr r s{L&#f% FC ff- ,,C { 55c:  & & ((ch[[55Vs/sHn[U5PM nnU$s snf)zQReturn a list of ByteTensor representing the random number states of all devices.)ranger r )iresultss r"r r -s/).|~)>?)>A}Q)>G? N@s/ new_statec^^[RR5 UR[RS9mSSS5 [ T[ 5(a[R"T5mO,[ T[5(a[R"ST5mUU4Sjn[U5 g!,(df  Ny=f)aSet 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_formatNrc>TRnUc [5n[RRUnUR T5 gN)rr rrr set_state)r r!rnew_state_copys r"cbset_rng_state..cbDs<ll ; "C!JJ99#>##N3r#) r_C_DisableFuncTorchclonecontiguous_formatrrrrr)r(rr/r.s ` @r"rr3s  # # %"u7N7NO &&#f% FC ff-4rN & %s B22 C new_statescB[U5Hup[X!5 M g)zSet the random number generator state of all devices. Args: new_states (Iterable of torch.ByteTensor): The desired state for each device. N) enumerater)r5r&states r"rrNs j)e*r#rc<^[T5mU4Sjn[USS9 g)amSet 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`. cv>[5n[RRUnUR T5 gr,)r rrrr)r r!rs r"r/manual_seed..cbgs.!JJ99#>%%d+r#T)rNrrrr/s` r"rrXs t9D, rr#c<^[T5mU4Sjn[USS9 g)zSet 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. c>[[55H1n[RRUnUR T5 M3 gr,)r%r rrrr)r&r!rs r"r/manual_seed_all..cbzs7|~&A % = =a @   ) )$ /'r#T)rNr<r=s` r"rros t9D0 rD!r#c Sn[U5 g)a\Set 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`. cr[5n[RRUnUR 5 gr,)r rrrrr r!s r"r/seed..cbs+!JJ99#> r#Nrr/s r"rrs! rNr#c Sn[U5 g)zSet 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. cSnSn[[55H\n[RRUnU(d$UR 5 UR 5nSnMKURU5 M^ g)NrFT)r%r rrrrrr) random_seedseededr&r!s r"r/seed_all..cbsb |~&A % = =a @ !&&(/<<> !--k:'r#NrErFs r"rrs ;rNr#c[5 [5n[RRUnUR 5$)zmReturn the current random seed of the current GPU. .. warning:: This function eagerly initializes CUDA. )r r rrrrrCs r"rrs5 L  C 55c:  ) ) ++r#)r)rN)typingrrrrrrr r r __all__rrrr r rrrrrrrr#r"rQs(( BB )%S%,, 67)V),4<@F$S#u||%;< 6 (6"2 t cd."#"$"&&, ,c ,r#