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) - 生成器的状态。