Python PyTorch Poisson用法及代碼示例

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

用法:

class torch.distributions.poisson.Poisson(rate, validate_args=None)

參數：

rate(數字,Tensor) -速率參數

基礎：torch.distributions.exp_family.ExponentialFamily

創建由速率參數 rate 參數化的泊鬆分布。

樣本是非負整數，pmf 由下式給出

\mathrm{rate}^k \frac{e^{-\mathrm{rate}}}{k!}

例子：

>>> m = Poisson(torch.tensor([4]))
>>> m.sample()
tensor([ 3.])