cupy.random.dirichlet#

cupy.random.dirichlet(alpha, size=None, dtype=<class 'float'>)[source]#

Dirichlet 分布。

返回从 dirichlet 分布中抽取的样本数组。其概率密度函数定义为

\[f(x) = \frac{\Gamma(\sum_{i=1}^K\alpha_i)} {\prod_{i=1}^{K}\Gamma(\alpha_i)} \prod_{i=1}^Kx_i^{\alpha_i-1}.\]
参数:
  • alpha (array) – dirichlet 分布的参数 \(\alpha\)

  • size (intintstuple) – 数组的形状。如果为 None,则生成一个零维数组。

  • dtype – 数据类型说明符。只允许使用 numpy.float32numpy.float64 类型。

返回:

从 dirichlet 分布中抽取的样本。

返回类型:

cupy.ndarray