Función PyTorch torch.set_rng_state

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.set_rng_stateEs una función en PyTorch utilizada para establecer el estado del generador de números aleatorios.

Definición de la función

torch.set_rng_state(new_state)

Descripción de parámetros

  • new_state- Nuevo estado del generador de números aleatorios (tipo Tensor)

Ejemplo de uso

Ejemplo

import torch

# Guardar el estado actual del generador de números aleatorios
saved_state = torch.get_rng_state()

# Establecer la semilla aleatoria
torch.manual_seed(123)

# Generar algunos números aleatorios
print("Primeros números aleatorios:", torch.rand(3))

# Restaurar el estado guardado anteriormente
torch.set_rng_state(saved_state)

# Volver a generar números aleatorios, deberían ser los mismos que antes
print("Números aleatorios después de restaurar:", torch.rand(3))

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones