StatelessSampleMultinomial

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	×
Atlas A2 训练系列产品/Atlas A2 推理系列产品	×
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	×
Atlas 训练系列产品	×

功能说明

• 算子功能：本算子是aclnnMultinomialGetWorkspaceSize和aclnnMultinomialTensorGetWorkspaceSize接口构建计算流程时使用的内部服务算子，用于Ascend 950场景下的有放回多项分布采样路径。算子根据输入的累积分布和随机种子，从每个多项分布中抽取num_samples个样本，并将样本的类别索引存储到输出张量中。

• 计算公式：

  对于第$d$个分布的第$j$次抽样，生成随机数：

  $$u_{d,j}\sim U(0,1]$$

  输出满足如下条件的最小类别索引$k$：

  $$x_{d,k-1}<u_{d,j}\le x_{d,k}$$

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	输入累积分布张量，shape为(C)或(N, C)，最后一维表示类别的累积概率。	FLOAT、FLOAT16、BFLOAT16	ND
norm_probs	可选输入	归一化概率张量，形状和数据类型与x相同。传入时用于采样结果的零概率类别回退。	FLOAT、FLOAT16、BFLOAT16	ND
seed	输入	随机数生成器的种子，影响生成的随机数序列。	INT64	ND
offset	输入	随机数生成器的偏移量，影响生成的随机数序列的位置。	INT64	ND
num_samples	属性	从每个多项分布中抽取的样本数。	INT64	-
y	输出	输出样本的类别索引，shape为(num_samples)或(N, num_samples)。	INT64	ND

约束说明

1. x为一维或二维张量，最后一维为类别维，其长度不超过$2^{24}$。
2. norm_probs为可选输入。传入时，其形状和数据类型与x一致，x为norm_probs沿最后一维计算累加和的结果；缺省时，通过比较x的相邻元素进行零概率类别回退。
3. num_samples为正整数。
4. offset为4的整数倍。

调用说明

调用方式	样例代码	说明
aclnn接口	test_aclnn_multinomial	通过aclnnMultinomial接口构建计算流程时，内部调用StatelessSampleMultinomial服务算子。
aclnn接口	test_aclnn_multinomial_tensor	通过aclnnMultinomialTensor接口构建计算流程时，内部调用StatelessSampleMultinomial服务算子。