np_random_state#

np_random_state(random_state_argument)[源码]#

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

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

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

参数：

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

返回：

_random_state函数: 其 random_state 关键字参数为 RandomState 实例的函数。

另见

py_random_state

示例

像这样装饰函数

@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)