多项式

多项式 - 22

版本

  • 名称: Multinomial (GitHub)

  • 域名: main

  • since_version: 22

  • 函数: False

  • support_level: SupportType.COMMON

  • 形状推断: True

此版本的运算符自版本22起可用。

摘要

根据每个可能结果的概率生成一个多项分布的样本张量。

属性

  • dtype - INT（默认值为 '6'）：

    （可选）输出张量元素的数据类型，如果未指定，我们将使用int32。

  • sample_size - INT (默认为 '1'):

    采样次数。

  • seed - FLOAT :

    （可选）随机生成器的种子，如果未指定，我们将自动生成一个。

输入

  • 输入 (异构) - T1:

    输入张量的形状为 [batch_size, class_size]，其中 class_size 是所有可能结果的数量。沿着零轴的每个值表示批次中每个对应结果的未归一化对数概率。

输出

  • 输出 (异构) - T2:

    输出张量的形状为 [batch_size, sample_size]，其中 sample_size 是采样的次数。沿着零轴的每个值表示批次中相应样本的结果。

类型约束

  • T1 在 ( tensor(bfloat16), tensor(double), tensor(float), tensor(float16) ) 中:

    将输入类型限制为浮点张量。

  • T2 在 ( tensor(int32), tensor(int64) ) 中:

    将输出类型限制为整数张量。

多项式 - 7

版本

  • 名称: Multinomial (GitHub)

  • 域名: main

  • since_version: 7

  • 函数: False

  • support_level: SupportType.COMMON

  • 形状推断: True

此版本的运算符自版本7起可用。

摘要

根据每个可能结果的概率生成一个多项分布的样本张量。

属性

  • dtype - INT（默认值为 '6'）：

    （可选）输出张量元素的数据类型，如果未指定，我们将使用int32。

  • sample_size - INT (默认为 '1'):

    采样次数。

  • seed - FLOAT :

    （可选）随机生成器的种子，如果未指定，我们将自动生成一个。

输入

  • 输入 (异构) - T1:

    输入张量的形状为 [batch_size, class_size]，其中 class_size 是所有可能结果的数量。沿着零轴的每个值表示批次中每个对应结果的未归一化对数概率。

输出

  • 输出 (异构) - T2:

    输出张量的形状为 [batch_size, sample_size]，其中 sample_size 是采样的次数。沿着零轴的每个值表示批次中相应样本的结果。

类型约束

  • T1 在 ( tensor(double), tensor(float), tensor(float16) ) 中:

    将输入类型限制为浮点张量。

  • T2 在 ( tensor(int32), tensor(int64) ) 中:

    将输出类型限制为整数张量。