torch_get_rng_state: RNG state management

View source: R/generator.R

torch_get_rng_stateR Documentation

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.