torch.cuda.get_rng_state
-
torch.cuda.get_rng_state(device='cuda')[source] -
Возвращает состояние генератора случайных чисел указанной видеокарты в виде объекта ByteTensor.
- Параметры
-
device (torch.device или int, необязательно) – Устройство, состояние генератора случайных чисел которого нужно вернуть. По умолчанию:
'cuda'(т.е.torch.device('cuda'), текущее устройство CUDA). - Тип возвращаемого значения
Предупреждение
Эта функция немедленно инициализирует CUDA.
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.cuda.get_rng_state.html
