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]