get_default_generator

get_default_generator#

jolideco.utils.torch.get_default_generator(device)[source]#

Get default torch generator

Parameters:

device (str) – Device name

Returns:

generator – Random number generator

Return type:

~torch.Generator