Python PyTorch Beta用法及代碼示例

本文簡要介紹python語言中 torch.distributions.beta.Beta 的用法。

用法:

class torch.distributions.beta.Beta(concentration1, concentration0, validate_args=None)

參數：

• concentration1(float或者Tensor) -分布的第一個濃度參數(通常稱為 alpha)

• concentration0(float或者Tensor) -分布的第二個濃度參數(通常稱為 beta)

基礎：torch.distributions.exp_family.ExponentialFamily

Beta 分布由 concentration1 和 concentration0 參數化。

例子：

>>> m = Beta(torch.tensor([0.5]), torch.tensor([0.5]))
>>> m.sample()  # Beta distributed with concentration concentration1 and concentration0
tensor([ 0.1046])