ott.neural.layers.PosDefPotentials.make_rng

ott.neural.layers.PosDefPotentials.make_rng#

PosDefPotentials.make_rng(name='default')#

Returns a new RNG key from a given RNG sequence for this Module.

The new RNG key is split from the previous one. Thus, every call to make_rng returns a new RNG key, while still guaranteeing full reproducibility.

TODO: Link to Flax RNG design note.

Parameters:

name (str) – The RNG sequence name.

Return type:

Array

Returns:

The newly generated RNG key.