StatelessSampleMultinomial

产品支持情况

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

功能说明

• 算子功能：本算子是aclnnMultinomialGetWorkspaceSize接口构建计算流程时使用的内部服务算子，用于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
seed	输入	随机数生成器的种子，影响生成的随机数序列。	INT64	ND
offset	输入	随机数生成器的偏移量，影响生成的随机数序列的位置。	INT64	ND
num_samples	属性	从每个多项分布中抽取的样本数。	INT64	-
y	输出	输出样本的类别索引，shape为(num_samples)或(N, num_samples)。	INT64	ND

约束说明

1. x仅支持1维或2维，最后一维C表示类别数，C不能超过2^24。
2. x需要表示有效的累积分布，最后一维应单调非递减。
3. num_samples必须大于0。
4. offset必须为4的倍数。
5. aclnn接口中replacement为false时，num_samples不能大于C。

调用说明

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