mindspore.ops.shuffle

View Source On Gitee
mindspore.ops.shuffle(x, seed=None)[source]

Randomly shuffle a tensor along its first dimension.

Parameters

  • x (Tensor) – The input tensor.

  • seed (int, optional) – Random seed. Default None , which is equivalent to 0.

Returns

Tensor

Supported Platforms:

Ascend GPU CPU

Examples

>>> import mindspore
>>> x = mindspore.tensor([1, 2, 3, 4], mindspore.float32)
>>> output = mindspore.ops.shuffle(x, seed=1)
>>> print(output)
[3. 4. 2. 1.]