Fonction torch.set_rng_state de PyTorch

Pytorch torch 参考手册Manuel de référence PyTorch torch

torch.set_rng_stateEst une fonction de PyTorch utilisée pour définir l'état du générateur de nombres aléatoires.

Définition de fonction

torch.set_rng_state(new_state)

Description du paramètre

  • new_state- Le nouvel état du générateur de nombres aléatoires (type Tensor)

Exemple d'utilisation

Exemple

import torch

# Enregistrer l'état actuel du générateur de nombres aléatoires
saved_state = torch.get_rng_state()

# Fixer la graine aléatoire
torch.manual_seed(123)

# Générer quelques nombres aléatoires
print("Premiers nombres aléatoires :", torch.rand(3))

# Restaurer l'état précédemment enregistré
torch.set_rng_state(saved_state)

# Régénérer les nombres aléatoires, ils devraient être identiques à ceux d'avant
print("Nombres aléatoires après restauration :", torch.rand(3))

Pytorch torch 参考手册Manuel de référence PyTorch torch

Autres extensions