PyTorch torch.wait-Funktion
PyTorch torch-Referenzhandbuch
torch.waitEs ist eine Funktion in PyTorch, die zum Warten verwendet wird. Sie wird verwendet, um CPU- und GPU-Operationen zu synchronisieren.
Funktionsdefinition
torch.wait(tensor)
Anwendungsbeispiel
Beispiel
import torch
# Erstelle einen Tensor (verwende GPU, wenn CUDA verfügbar ist)
if torch.cuda.is_available():
x = torch.tensor([1, 2, 3]).cuda()
torch.wait(x)
print("Warten auf CUDA-Tensor abgeschlossen")
else:
x = torch.tensor([1, 2, 3])
torch.wait(x)
print("Warten auf CPU-Tensor abgeschlossen")
# Erstelle einen Tensor (verwende GPU, wenn CUDA verfügbar ist)
if torch.cuda.is_available():
x = torch.tensor([1, 2, 3]).cuda()
torch.wait(x)
print("Warten auf CUDA-Tensor abgeschlossen")
else:
x = torch.tensor([1, 2, 3])
torch.wait(x)
print("Warten auf CPU-Tensor abgeschlossen")
Weitere Erweiterungen