np_random_state#

np_random_state(random_state_argument)[source]#

装饰器用于生成一个 numpy 的 RandomState 或 Generator 实例。

装饰器使用 nx.utils.create_random_state() 处理由 random_state_argument 指示的参数。参数值可以是一个种子（整数），或者是一个 numpy.random.RandomState 或 numpy.random.RandomState 实例，或者（ None 或 numpy.random ）。后两种选项使用 numpy.random 的全局随机数生成器。

返回的实例是一个 numpy.random.RandomState 或 numpy.random.Generator 。

Parameters:

random_state_argument字符串或整数: 要转换为 numpy.random.RandomState 实例的参数的名称或索引。

Returns:

_random_state函数: 其 random_state 关键字参数是一个 RandomState 实例的函数。

See also

py_random_state

Examples

按如下方式装饰函数:

@np_random_state("seed")
def random_float(seed=None):
    return seed.rand()

@np_random_state(0)
def random_float(rng=None):
    return rng.rand()

@np_random_state(1)
def random_array(dims, random_state=1):
    return random_state.rand(*dims)