mindspore.ops.Randperm

class mindspore.ops.Randperm(max_length=1, pad=- 1, dtype=mstype.int32)[源代码]

生成从0到n-1不重复的n个随机样本。如果 max_length > n，则末尾的 max_length-n 个元素使用 pad 填充。

参数：

  • max_length (int) - 取样数量，必须大于0。默认值：1。

  • pad (int) - 填充值。默认值：-1。

  • dtype (mindspore.dtype) - 输出的数据类型。默认值：mindspore.int32。

输入：

  • n (Tensor[int32]) - shape为 \((1,)\) 的输入Tensor，须在[0, max_length]内取值。

输出：

  • output (Tensor) - shape: (max_length,)，数据类型为 dtype 。

异常：

  • TypeError - max_length 或 pad 不是int类型。

  • TypeError - n 不是Tensor。

  • TypeError - n 包含非int元素。

  • TypeError - n 包含负数。

支持平台：

Ascend GPU

样例：

>>> # The result of every execution is different because this operator will generate n random samples.
>>> randperm = ops.Randperm(max_length=30, pad=-1)
>>> n = Tensor([20], dtype=mindspore.int32)
>>> output = randperm(n)
>>> print(output)
[15 6 11 19 14 16 9 5 13 18 4 10 8 0 17 2 1 12 3 7
 -1 -1 -1 -1 -1 -1 -1 -1 -1 -1]