torch.cuda.get_rng_state

torch.cuda.get_rng_state(device='cuda') [source]

Возвращает состояние генератора случайных чисел указанной видеокарты в виде тензора ByteTensor.

Параметры:: device (torch.device или int, необязательно) – Устройство, состояние RNG которого требуется вернуть. По умолчанию: '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/1.13/generated/torch.cuda.get_rng_state.html