mindspore.ops.RandpermV2

class mindspore.ops.RandpermV2(dtype=mstype.int64)[source]

Generates random permutation of integers from 0 to n-1 without repeating.

Refer to mindspore.ops.randperm() for more details.

Warning

This is an experimental API that is subject to change or deletion.

Parameters

dtype (mindspore.dtype, optional) – The type of output. Its value must be one of the following types: int32, int16, int8, uint8, int64, float64, float32, float16. Default: mstype.int64.

Inputs:
  • n (Union[Tensor, int]) - The input n Tensor with shape \(()\) or \((1,)\) and with data type of int64.

  • seed (int, optional) - Random seed. Default: 0 . When seed is -1 (only negative value), offset is 0, it’s determined by time.

  • offset (int, optional) - Offset to generate random numbers. Priority is higher than random seed. Default: 0 . It must be non-negative.

Outputs:

Tensor. Its shape is specified by the required args n. Its type is specified by dtype. Otherwise is default.

Supported Platforms:

Ascend CPU

Examples

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