多项式分布¶
多项式分布 - 22¶
版本¶
名称: 多项式分布 (GitHub)
域:
main自版本:
22函数:
False支持级别:
SupportType.COMMON形状推断:
True
此版本的算子已自版本 22起可用。
摘要¶
根据每个可能结果的概率生成来自多项式分布的样本张量。
属性¶
dtype - INT（默认值为
'6'）（可选）输出张量的元素的数据类型，如果未指定，我们将使用 int32。
sample_size - INT（默认值为
'1'）采样次数。
seed - FLOAT :
（可选）随机生成器的种子，如果未指定，我们将自动生成一个。
输入¶
input (异构) - T1
输入张量，形状为 [batch_size, class_size]，其中 class_size 是所有可能结果的数量。沿轴零的每个值表示批次中每个对应结果的未归一化对数概率。
输出¶
output (异构) - T2
输出张量，形状为 [batch_size, sample_size]，其中 sample_size 是采样次数。沿轴零的每个值表示批次中对应样本的结果。
类型约束¶
T1 in (
tensor(bfloat16),tensor(double),tensor(float),tensor(float16))将输入类型约束为浮点张量。
T2 in (
tensor(int32),tensor(int64))将输出类型约束为整型张量。
多项式分布 - 7¶
版本¶
名称: 多项式分布 (GitHub)
域:
main自版本:
7函数:
False支持级别:
SupportType.COMMON形状推断:
True
此版本的算子已自版本 7起可用。
摘要¶
根据每个可能结果的概率生成来自多项式分布的样本张量。
属性¶
dtype - INT（默认值为
'6'）（可选）输出张量的元素的数据类型，如果未指定，我们将使用 int32。
sample_size - INT（默认值为
'1'）采样次数。
seed - FLOAT :
（可选）随机生成器的种子，如果未指定，我们将自动生成一个。
输入¶
input (异构) - T1
输入张量，形状为 [batch_size, class_size]，其中 class_size 是所有可能结果的数量。沿轴零的每个值表示批次中每个对应结果的未归一化对数概率。
输出¶
output (异构) - T2
输出张量，形状为 [batch_size, sample_size]，其中 sample_size 是采样次数。沿轴零的每个值表示批次中对应样本的结果。
类型约束¶
T1 in (
tensor(double),tensor(float),tensor(float16))将输入类型约束为浮点张量。
T2 in (
tensor(int32),tensor(int64))将输出类型约束为整型张量。