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)
)将输出类型约束为整型张量。