Функция PyTorch torch.cuda.seed
Справочное руководство Pytorch torch
torch.cuda.seedЭто функция в PyTorch, используемая для установки начального числа генератора случайных чисел CUDA. Устанавливает начальное число генератора случайных чисел для текущего устройства CUDA (функционально эквивалентна manual_seed).
Определение функции
torch.cuda.seed()
Пример использования
Пример
import torch
# Установить начальное число генератора случайных чисел CUDA
if torch.cuda.is_available():
torch.cuda.seed()
x = torch.randn(3, 3).cuda()
print(f"Случайный тензор на CUDA: {x}")
else:
print("CUDA недоступна")
# Установить начальное число генератора случайных чисел CUDA
if torch.cuda.is_available():
torch.cuda.seed()
x = torch.randn(3, 3).cuda()
print(f"Случайный тензор на CUDA: {x}")
else:
print("CUDA недоступна")
Другие расширения