Multinomial

功能

返回Multinomial采样结果矩阵。

输入

一个输入：

x：tensor，shape=[batch_size, class_size]，数据类型：float16、float。class_size 指所有可能结果的数量。每个值表示该batch中每个相应结果的非归一化对数概率。

输出

一个输出：

y：一个tensor，shape=[batch_size, sample_size]，输出数据类型是int32、int64。sample_size指采样的次数。每个值表示该batch中相应样本的结果。

属性

dtype：int，默认为6，含义：输出数据类型，默认为int32。

sample_size：int，默认为1，含义：采样次数。

seed：float，随机数种子。

支持的ONNX版本

Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18