mindspore.mint.rand

查看源文件
mindspore.mint.rand(*size, generator=None, dtype=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,必须是float类型。如果是None,mindspore.float32 会被使用。默认值: None

返回:

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

异常:
  • ValueError - 如果 dtype 不是一个 mstype.float_type 类型。

支持平台:

Ascend

样例:

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