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是输入 shape 、 alpha 拼接后的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)