Multinomial
功能
返回Multinomial采样结果矩阵
边界
【输入】
1个输入
x:tensor,shape=[batch_size, class_size],数据类型:float16、float
【输出】
一个输出
y:一个张量,shape=[batch_size, sample_size],输出数据类型是int32、int64
【属性】
dtype:int,默认为6,含义:输出数据类型,默认为int32
sample_size:int,默认为1,含义:采样次数
seed:float,随机数种子

如果调用该算子超时,需要使用ret = acl.rt.set_op_execute_time_out(timeout)接口避免超时。
支持的ONNX版本
Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18
父主题: 支持ONNX算子清单