Python PyTorch Gamma用法及代碼示例

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

用法:

class torch.distributions.gamma.Gamma(concentration, rate, validate_args=None)

參數：

• concentration(float或者Tensor) -分布的形狀參數(通常稱為 alpha)

• rate(float或者Tensor) -rate = 1 /分布規模(通常稱為 beta)

基礎：torch.distributions.exp_family.ExponentialFamily

創建由形狀 concentration 和 rate 參數化的 Gamma 分布。

例子：

>>> m = Gamma(torch.tensor([1.0]), torch.tensor([1.0]))
>>> m.sample()  # Gamma distributed with concentration=1 and rate=1
tensor([ 0.1046])