多项式分布¶
多项式分布 - 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))将输出类型限制为整数张量。