pytorch 之 manual_seed

 La fuente de pytorch manual_seed sigue, el papel de: generar un número aleatorio fijo

def manual_seed(seed):
    r"""Sets the seed for generating random numbers. Returns a
    `torch.Generator` object.

    Args:
        seed (int): The desired seed.
    """
    seed = int(seed)
    import torch.cuda

    if not torch.cuda._in_bad_fork:
        torch.cuda.manual_seed_all(seed)

    return default_generator.manual_seed(seed)

Ejemplos:

import torch
print(torch.rand(2))
torch.manual_seed(2)

print(torch.rand(2))

Ejecutar varias veces después de conocer los resultados