mindspore.mint.rand

查看源文件
mindspore.mint.rand(*size, generator=None, dtype=None, device=None)[源代码]

返回一个Tensor，shape和dtype由输入决定，其元素为服从均匀分布的 \([0, 1)\) 区间的数字。

参数：

  • size (Union[int, tuple(int), list(int)]) - 输出的Tensor的shape，例如，\((2, 3)\) or \(2\)。

关键字参数：

  • generator (mindspore.Generator, 可选) - 伪随机数生成器。默认值： None ，使用默认伪随机数生成器。

  • dtype (mindspore.dtype，可选) - 指定的输出Tensor的dtype。如果是 None ， mindspore.float32 会被使用。默认值： None 。

  • device (str, 可选) - 指定Tensor使用的内存来源。仅支持 "Ascend" 、 "npu" 。如果为 None ，那么将会使用 mindspore.set_device() 设置的值。默认值： None 。

返回：

Tensor，shape和dtype由输入决定其元素为服从均匀分布的 \([0, 1)\) 区间的数字。

异常：

  • ValueError - 如果 size 包含负数。

  • ValueError - 如果 device 是 "GPU" 。

  • RuntimeError - 如果 device 是 "CPU" 。

支持平台：

Ascend

样例：

>>> from mindspore import mint
>>> print(mint.rand(2, 3).shape)
(2, 3)