mindspore.Generator

class mindspore.Generator

管理随机数状态的生成器，为随机函数提供seed和offset。当seed和offset固定时，随机函数产生相同的随机序列。

输入：

• step (int) - 偏移量更新的步长。

输出：

包含生成器的种子和偏移量的tuple。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore as ms
>>> from mindspore import Generator
>>> generator = Generator()
>>> generator.manual_seed(5)
>>> print(generator.initial_seed())
5
>>> state = generator.get_state()
>>> generator.seed()
>>> generator.set_state(state)
>>> print(generator.initial_seed())
5

get_state()

获取生成器状态。

返回：

Tensor，生成器的状态。

initial_seed()

返回生成器的初始种子。

返回：

生成器的初始化种子。

manual_seed(seed)

设置生成器种子。

参数：

• seed (int) - 生成器的种子。

返回：

生成器自身。

seed()

生成可作为生成器种子的随机种子。

返回：

int类型，随机生成的种子。

set_state(state)

设置生成器状态。

参数：

• state (tensor) - 生成器的状态。