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