RNG state management — torch_get_rng_state • torch

Low level functionality to set and change the RNG state. It's recommended to use torch_manual_seed() for most cases.

Usage

torch_get_rng_state()

torch_set_rng_state(state)

cuda_get_rng_state(device = NULL)

cuda_set_rng_state(state, device = NULL)

Arguments

state: A tensor with the current state or a list containing the state for each device - (for CUDA).
device: The cuda device index to get or set the state. If NULL gets the state for all available devices.

Functions

torch_set_rng_state(): Sets the RNG state for the CPU
cuda_get_rng_state(): Gets the RNG state for CUDA.
cuda_set_rng_state(): Sets the RNG state for CUDA.