PyTorch función torch.wait
Manual de referencia de Pytorch torch
torch.waitEs una función utilizada para esperar en PyTorch. Se utiliza para sincronizar operaciones de CPU y GPU.
Definición de la función
torch.wait(tensor)
Ejemplo de uso
Ejemplo
import torch
# Crear tensor (usar GPU si CUDA está disponible)
if torch.cuda.is_available():
x = torch.tensor([1, 2, 3]).cuda()
torch.wait(x)
print("Wait completed on CUDA tensor")
else:
x = torch.tensor([1, 2, 3])
torch.wait(x)
print("Wait completed on CPU tensor")
# Crear tensor (usar GPU si CUDA está disponible)
if torch.cuda.is_available():
x = torch.tensor([1, 2, 3]).cuda()
torch.wait(x)
print("Wait completed on CUDA tensor")
else:
x = torch.tensor([1, 2, 3])
torch.wait(x)
print("Wait completed on CPU tensor")
Otras extensiones