PyTorch torch.set_rng_state Funktion
PyTorch torch-Referenzhandbuch
torch.set_rng_stateIst eine Funktion in PyTorch, mit der der Zustand des Zufallszahlengenerators festgelegt wird.
Funktionsdefinition
torch.set_rng_state(new_state)
Parameterbeschreibung
new_state- Der neue Zustand des Zufallszahlengenerators (Tensor-Typ)
Anwendungsbeispiel
Beispiel
import torch
# Aktuellen Zustand des Zufallszahlengenerators speichern
saved_state = torch.get_rng_state()
# Zufallsstartwert festlegen
torch.manual_seed(123)
# Einige Zufallszahlen generieren
print("Erste Zufallszahlen:", torch.rand(3))
# Den zuvor gespeicherten Zustand wiederherstellen
torch.set_rng_state(saved_state)
# Zufallszahlen erneut generieren, sie sollten dieselben wie zuvor sein
print("Zufallszahlen nach Wiederherstellung:", torch.rand(3))
# Aktuellen Zustand des Zufallszahlengenerators speichern
saved_state = torch.get_rng_state()
# Zufallsstartwert festlegen
torch.manual_seed(123)
# Einige Zufallszahlen generieren
print("Erste Zufallszahlen:", torch.rand(3))
# Den zuvor gespeicherten Zustand wiederherstellen
torch.set_rng_state(saved_state)
# Zufallszahlen erneut generieren, sie sollten dieselben wie zuvor sein
print("Zufallszahlen nach Wiederherstellung:", torch.rand(3))
Andere Erweiterungen