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])