mindspore.ops.multinomial_with_replacement

mindspore.ops.multinomial_with_replacement(x, seed, offset, numsamples, replacement=False)

生成一个多项式分布的tensor。

说明

• 输入的行不需要求和为1（在这种情况下，使用值作为权重），但必须是非负的、有限的，并且具有非零和。

• seed 如果为 -1 且 offset 为 0 ，则随机数生成器将使用随机种子。

参数：

• x (Tensor) - 一维或二维的输入tensor，包含概率的累积和。

• seed (int) - 随机种子。

• offset (int) - 偏移量。

• numsamples (int) - 采样的次数。

• replacement (bool，可选) - 是否放回。默认 False 。

返回：

Tensor

支持平台：

CPU

样例：

>>> import mindspore
>>> x = mindspore.tensor([[0., 9., 4., 0.]], mindspore.float32)
>>> mindspore.ops.multinomial_with_replacement(x, 2, 5, 2, True)
Tensor(shape=[1, 2], dtype=Int64, value=
[[1, 1]])