mindspore.ops.random_gamma

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

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

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

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

  • seed (int, 可选) - 随机数生成器的种子,必须是非负数,默认为 None

返回:

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

异常:
  • TypeError - shape 不是Tensor。

  • TypeError - alpha 不是Tensor。

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

  • TypeError - 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)