mindspore.numpy.rand

查看源文件
mindspore.numpy.rand(*shape, dtype=mstype.float32)[源代码]

返回一个给定shape和类型的新Tensor,其中所有元素以区间 \([0,1)\) 上均匀分布的随机数来填充。

参数:
  • shape (Union[int, tuple(int), list(int)]) - 指定新Tensor的shape,例如 \((2,3)\)\(2\)

  • dtype (Union[mindspore.dtype, str], 可选) - 指定Tensor的数据类型。默认值: mstype.float32

返回:

Tensor,给定shape和类型,其中所有元素都为区间 \([0,1)\) 上均匀分布的随机数。

异常:
  • TypeError - 如果输入参数非上述给定的类型。

  • ValueError - 如果 dtype 不是float类型。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore.numpy as np
>>> from mindspore import set_seed
>>> set_seed(1)
>>> print(np.rand((2,3)))
[[4.1702199e-01 9.9718481e-01 7.2032452e-01]
[9.3255734e-01 1.1438108e-04 1.2812445e-01]]