采样功能。对最后一个轴进行采样，随机抽取numSamples个值，输出下标。

struct MultinomialParam {
    uint32_t numSamples = 1;
    uint32_t randSeed = 0;
};

成员名称	类型	默认值	描述
numSamples	uint32_t	1	随机采样数。 numSamples应小于等于x对应的维度大小，最大为64。
randSeed	uint32_t	0	随机数种子。

成员名称

类型

默认值

描述

numSamples

uint32_t

随机采样数。

numSamples应小于等于x对应的维度大小，最大为64。

randSeed

uint32_t

随机数种子。

参数	维度	数据类型	格式	描述
x	[batch, vocSize]	float16	ND	输入tensor。

参数	维度	数据类型	格式	描述
output	[batch, numSamples]	int32	ND	采数数据下标。dims[1]必须等于numSamples

用户需确保对最后一个轴进行归一化操作。

输入

intensor =

[[0.0701, 0.1021, 0.1010, 0.1005],
[0.1101, 0.1020, 0.1017, 0.0990],
[0.0592, 0.1006, 0.1006, 0.1002]]

numSamples = 1

randSeed = 0

MultinomialOperation