mindspore.ops.shuffle

查看源文件
mindspore.ops.shuffle(x, seed=None)[源代码]

沿着第一维随机打乱tensor。

参数:
  • x (Tensor) - 输入tensor。

  • seed (int, 可选) - 随机种子,默认 None ,等同于 0

返回:

Tensor

支持平台:

Ascend GPU CPU

样例:

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