randperm

paddle. randperm ( n, dtype='int64', name=None ) [源代码]

返回一个数值在 0 到 n-1、随机排列的 1-D Tensor，数据类型为 dtype。

参数

  • n (int) - 随机序列的上限（不包括在序列中），应该大于 0。

  • dtype (str|np.dtype，可选) - 输出 Tensor 的数据类型，支持 int32、int64、float32、float64。默认值为 int64。

  • name (str，可选) - 具体用法请参见 Name，一般无需设置，默认值为 None。

返回

Tensor：一个数值在 0 到 n-1、随机排列的 1-D Tensor，数据类型为 dtype 。

代码示例

import paddle

out1 = paddle.randperm(5)
# [4, 1, 2, 3, 0]  # random

out2 = paddle.randperm(7, 'int32')
# [1, 6, 2, 0, 4, 3, 5]  # random

使用本API的教程文档