mindspore.ops.random_gamma

mindspore.ops.random_gamma(shape, alpha, seed=0, seed2=0)[源代码]

根据伽马分布产生成随机数。

参数：

  • shape (Tensor) - 指定生成随机数的shape。任意维度的Tensor。

  • alpha (Tensor) - \(\alpha\) 分布的参数。应该大于0且数据类型为half、float32或者float64。

  • seed (int) - 随机数生成器的种子，必须是非负数，默认为0。

  • seed2 (int) - 随机数生成器的种子，必须是非负数，默认为0。

返回：

Tensor。shape是输入 shape 、 alpha 拼接后的shape。数据类型和alpha一致。

异常：

  • TypeError C shape 不是Tensor。

  • TypeError C alpha 不是Tensor。

  • TypeError C seed 的数据类型不是int。

  • TypeError C alpha 的数据类型不是half、float32或者float64。

支持平台：

CPU

样例：

>>> import numpy as np
>>> import mindspore
>>> from mindspore import Tensor, ops
>>> shape = Tensor(np.array([7, 5]), mindspore.int32)
>>> alpha = Tensor(np.array([0.5, 1.5]), mindspore.float32)
>>> output = ops.random_gamma(shape, alpha, seed=5)
>>> result = output.shape
>>> print(result)
(7, 5, 2)