Fonction torch.wait de PyTorch
Manuel de référence PyTorch torch
torch.waitC'est une fonction de PyTorch utilisée pour attendre. Elle sert à synchroniser les opérations CPU et GPU.
Définition de fonction
torch.wait(tensor)
Exemple d'utilisation
Exemple
import torch
# Créer un tenseur (utiliser le GPU si CUDA est disponible)
if torch.cuda.is_available():
x = torch.tensor([1, 2, 3]).cuda()
torch.wait(x)
print("Attente terminée sur le tenseur CUDA")
else:
x = torch.tensor([1, 2, 3])
torch.wait(x)
print("Attente terminée sur le tenseur CPU")
# Créer un tenseur (utiliser le GPU si CUDA est disponible)
if torch.cuda.is_available():
x = torch.tensor([1, 2, 3]).cuda()
torch.wait(x)
print("Attente terminée sur le tenseur CUDA")
else:
x = torch.tensor([1, 2, 3])
torch.wait(x)
print("Attente terminée sur le tenseur CPU")
Autres extensions