Model.set_seed()

Contents

Model.set_seed()#

Model.set_seed(seed)[source]#

Splits and sets the seed / PRNG key.

Parameters:: seed – The seed is split and distributed to the seed nodes of the model. Must be a KeyArray, i.e. an array of shape (2,) and dtype uint32. See jax.random for more details.