Multinomial

Multinomial - 22

版本

  • **名称**: Multinomial (GitHub)

  • **域**: main

  • **起始版本**: 22

  • **函数**: False

  • **支持级别**: 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) )

    将输出类型约束为整型张量。

Multinomial - 7

版本

  • **名称**: Multinomial (GitHub)

  • **域**: main

  • **起始版本**: 7

  • **函数**: False

  • **支持级别**: 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) )

    将输出类型约束为整型张量。