多项式分布

多项式分布 - 22

版本

  • 名称: 多项式分布 (GitHub)

  • : main

  • 起始版本: 22

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

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

摘要

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

属性

  • dtype - INT (默认值为 '6')

    (可选)输出张量元素的Dtype,如果未指定,我们将使用int32。

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

    采样次数。

  • 种子 - 浮点型 :

    (可选)随机生成器的种子,如果未指定,我们将自动生成一个。

输入

  • input (异构) - T1

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

输出

  • output (异构) - T2

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

类型约束

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

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

  • T2 包含 ( tensor(int32), tensor(int64) )

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

多项式分布 - 7

版本

  • 名称: 多项式分布 (GitHub)

  • : main

  • 起始版本: 7

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的操作符自版本 7 起可用。

摘要

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

属性

  • dtype - INT (默认值为 '6')

    (可选)输出张量元素的Dtype,如果未指定,我们将使用int32。

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

    采样次数。

  • 种子 - 浮点型 :

    (可选)随机生成器的种子,如果未指定,我们将自动生成一个。

输入

  • input (异构) - T1

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

输出

  • output (异构) - T2

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

类型约束

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

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

  • T2 包含 ( tensor(int32), tensor(int64) )

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