Poisson
Source:R/gen-namespace-docs.R, R/gen-namespace-examples.R, R/gen-namespace.R
torch_poisson.RdPoisson
Arguments
- self
(Tensor) the input tensor containing the rates of the Poisson distribution
- generator
(
torch.Generator, optional) a pseudorandom number generator for sampling
poisson(input *, generator=NULL) -> Tensor
Returns a tensor of the same size as input with each element
sampled from a Poisson distribution with rate parameter given by the corresponding
element in input i.e.,
$$ \mbox{out}_i \sim \mbox{Poisson}(\mbox{input}_i) $$
Examples
if (torch_is_installed()) {
rates = torch_rand(c(4, 4)) * 5 # rate parameter between 0 and 5
torch_poisson(rates)
}
#> torch_tensor
#> 2 2 4 3
#> 3 3 1 4
#> 1 2 6 1
#> 4 1 2 4
#> [ CPUFloatType{4,4} ]