You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

mindspore.ops.Gamma.rst 1.6 kB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. mindspore.ops.Gamma
  2. ===================
  3. .. py:class:: mindspore.ops.Gamma(*args, **kwargs)
  4. 根据概率密度函数分布生成随机正浮点数x。
  5. .. math::
  6. \text{P}(x|α,β) = \frac{\exp(-x/β)}{{β^α}\cdot{\Gamma(α)}}\cdot{x^{α-1}}
  7. **参数:**
  8. - **seed** (int) - 算子层的随机种子,用于生成随机数。必须是非负的。默认值:0。
  9. - **seed2** (int):全局的随机种子,和算子层的随机种子共同决定最终生成的随机数。必须是非负的。默认值:0。
  10. **输入:**
  11. - **shape** (tuple) - 待生成的随机Tensor的shape。只支持常量值。
  12. - **alpha** (Tensor) - α为Gamma分布的shape parameter,主要决定了曲线的形状。其值必须大于0。数据类型为float32。
  13. - **beta** (Tensor) - β为Gamma分布的inverse scale parameter,主要决定了曲线有多陡。其值必须大于0。数据类型为float32。
  14. **输出:**
  15. Tensor。shape是输入 `shape` 以及alpha、beta广播后的shape。数据类型为float32。
  16. **异常:**
  17. - **TypeError** - `seed` 和 `seed2` 都不是int。
  18. - **TypeError** - `alpha` 和 `beta` 都不是Tensor。
  19. - **ValueError** - `shape` 不是常量值。
  20. **支持平台:**
  21. ``Ascend``
  22. **样例:**
  23. >>> shape = (3, 1, 2)
  24. >>> alpha = Tensor(np.array([[3, 4], [5, 6]]), mstype.float32)
  25. >>> beta = Tensor(np.array([1.0]), mstype.float32)
  26. >>> gamma = ops.Gamma(seed=3)
  27. >>> output = gamma(shape, alpha, beta)
  28. >>> result = output.shape
  29. >>> print(result)
  30. (3, 2, 2)