rng# fkat.utils.rng.get_rng_states() → dict[str, Any][source]# Collect the global random state of torch, torch.cuda, numpy and Python. fkat.utils.rng.set_rng_states(rng_state_dict: dict[str, Any]) → None[source]# Set the global random state of torch, torch.cuda, numpy and Python in the current process.