set_rng_state

torch.tvarant.set_rng_state(new_state, device='tvarant') → None

Restores RNG state previously returned by get_rng_state.

Restore RNG state previously returned by get_rng_state.

Parameters:

Name	Type	Description	Default
new_state	Tensor	State tensor from get_rng_state.	required
device	Union[int, str, device]	Device specifier (compatibility; ignored beyond validation).	'tvarant'

Source code in torch_tvarant/tvarant/__init__.py

def set_rng_state(
    new_state: torch.Tensor, device: Union[int, str, torch.device] = "tvarant"
) -> None:
    """Restore RNG state previously returned by ``get_rng_state``.

    Args:
        new_state: State tensor from ``get_rng_state``.
        device: Device specifier (compatibility; ignored beyond validation).
    """
    _ = device
    _C.set_rng_state(new_state)