mindspore.ops.randperm

查看源文件
mindspore.ops.randperm(n, seed=0, offset=0, dtype=mstype.int64)[源代码]

生成从 0 到 n-1 的整数随机排列。

警告

  • 这是一个实验性API,后续可能修改或删除。

  • Ascend后端不支持随机数重现功能, seed 参数不起作用。

参数:
  • n (Union[Tensor, int]) - 输入上界(不包含)。

  • seed (int, 可选) - 随机种子。默认 0 。当seed为-1时,offset为0,随机数由时间决定。

  • offset (int, 可选) - 偏移量。生成随机数,优先级高于随机种子。必须是非负数,默认 0

  • dtype (mindspore.dtype, 可选) - 指定数据类型。默认 mstype.int64

返回:

Tensor,shape由参数 n 决定。

支持平台:

CPU

样例:

>>> import mindspore
>>> n, seed, offset = 4, 0, 0
>>> output = mindspore.ops.randperm(n, seed, offset, dtype=mindspore.int64)
>>> print(output)
[0 2 1 3]