Python PyTorch poisson用法及代碼示例

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

用法:

torch.poisson(input, generator=None) → Tensor

參數：

input(Tensor) -包含泊鬆分布率的輸入張量

關鍵字參數：

generator(torch.Generator， 可選的) -用於采樣的偽隨機數發生器

返回與input 大小相同的張量，每個元素從泊鬆分布中采樣，速率參數由input 中的相應元素給出，即

\text{out}_i \sim \text{Poisson}(\text{input}_i)

例子：

>>> rates = torch.rand(4, 4) * 5  # rate parameter between 0 and 5
>>> torch.poisson(rates)
tensor([[9., 1., 3., 5.],
        [8., 6., 6., 0.],
        [0., 4., 5., 3.],
        [2., 1., 4., 2.]])