poisson4

nvmath.device.random.poisson4(state, Lambda)

使用指定的位生成器状态，从泊松分布中采样四个 uint。

参数:

• state – 对应于其中一个位生成器的状态对象。

• Lambda – 表征泊松分布的参数。

返回值:

类型为 nvmath.device.uint32x4 的值。

示例

下面内核的 states 参数应该是一个已初始化的位生成器状态数组。 有关如何创建和初始化位生成器状态的更多详细信息，请参阅 nvmath.device.random.init() 的文档。

>>> from numba import cuda
>>> from nvmath.device import random
>>> compiled_apis = random.Compile()
>>> @cuda.jit(link=compiled_apis.files, extensions=compiled_apis.extension)
... def kernel(states, Lambda):
...     i = cuda.grid(1)
...     value = random.poisson4(states[i], Lambda)