torch.accelerator.random.get_rng_state_all

torch.accelerator.random.get_rng_state_all()

Return a list of torch.Tensor of dtype torch.uint8 representing the RNG states of all devices for the current accelerator.

Returns:

the RNG states of the default generators for all devices.

Return type:

list[torch.Tensor]

Warning

This function eagerly initializes the accelerator runtime.