Функция torch.wait в PyTorch
Справочное руководство PyTorch torch
torch.waitЭто функция в PyTorch, используемая для ожидания. Она используется для синхронизации операций CPU и GPU.
Определение функции
torch.wait(tensor)
Пример использования
Пример
import torch
# Создать тензор (использовать GPU, если CUDA доступна)
if torch.cuda.is_available():
x = torch.tensor([1, 2, 3]).cuda()
torch.wait(x)
print("Ожидание завершено на тензоре CUDA")
else:
x = torch.tensor([1, 2, 3])
torch.wait(x)
print("Ожидание завершено на тензоре CPU")
# Создать тензор (использовать GPU, если CUDA доступна)
if torch.cuda.is_available():
x = torch.tensor([1, 2, 3]).cuda()
torch.wait(x)
print("Ожидание завершено на тензоре CUDA")
else:
x = torch.tensor([1, 2, 3])
torch.wait(x)
print("Ожидание завершено на тензоре CPU")
Другие расширения