Python PyTorch Dirichlet用法及代码示例

本文简要介绍python语言中 torch.distributions.dirichlet.Dirichlet 的用法。

用法:

class torch.distributions.dirichlet.Dirichlet(concentration, validate_args=None)

参数：

concentration(Tensor) -分布的浓度参数(通常称为 alpha)

基础：torch.distributions.exp_family.ExponentialFamily

创建由浓度 concentration 参数化的狄利克雷分布。

例子：

>>> m = Dirichlet(torch.tensor([0.5, 0.5]))
>>> m.sample()  # Dirichlet distributed with concentrarion concentration
tensor([ 0.1046,  0.8954])