PyTorch torch.cuda.seed Funktion
PyTorch torch Referenzhandbuch
torch.cuda.seedIst eine Funktion in PyTorch, mit der der CUDA-Zufallsseed festgelegt wird. Legt den Zufallsseed für das aktuelle CUDA-Gerät fest (funktional identisch mit manual_seed).
Funktionsdefinition
torch.cuda.seed()
Anwendungsbeispiel
Beispiel
import torch
# CUDA-Zufallsseed setzen
if torch.cuda.is_available():
torch.cuda.seed()
x = torch.randn(3, 3).cuda()
print(f"Zufälliger Tensor auf CUDA: {x}")
else:
print("CUDA nicht verfügbar")
# CUDA-Zufallsseed setzen
if torch.cuda.is_available():
torch.cuda.seed()
x = torch.randn(3, 3).cuda()
print(f"Zufälliger Tensor auf CUDA: {x}")
else:
print("CUDA nicht verfügbar")
Andere Erweiterungen