torch_get_rng_state: RNG state management

In torch: Tensors and Neural Networks with 'GPU' Acceleration

RNG state management

Description

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.

torch documentation built on May 29, 2024, 9:54 a.m.