coerce_prng¶

torch_sim.state.coerce_prng(rng, device)[source]¶

Coerce an int seed or existing Generator into a torch.Generator.

Parameters:

rng (int | Generator | None) – An int seed, an existing torch.Generator, or None for unseeded.
device (str | device | int | None) – Target device for the returned generator.

Returns:

A torch.Generator on device.

Return type:

Generator