全局平均池化¶
全局平均池化 - 22¶
版本¶
名称: 全局平均池化 (GitHub)
域:
main
自版本:
22
函数:
False
支持级别:
SupportType.COMMON
形状推断:
True
此版本的运算符自**版本 22**起可用。
摘要¶
全局平均池化使用输入张量 X,并在同一通道中的值上应用平均池化。这等效于内核大小等于输入张量空间维度的平均池化。
输入¶
X (异构) - T
来自先前运算符的输入数据张量;图像情况下的维度为 (N x C x H x W),其中 N 是批大小,C 是通道数,H 和 W 是数据的高度和宽度。对于非图像情况,维度采用 (N x C x D1 x D2 … Dn) 的形式,其中 N 是批大小。
输出¶
Y (异构) - T
跨输入张量池化的输出数据张量。输出张量的秩与输入相同。输出形状的前两个维度与输入相同 (N x C),而其他维度均为 1。
类型约束¶
T in (
tensor(bfloat16)
,tensor(double)
,tensor(float)
,tensor(float16)
)将输入和输出类型约束为浮点张量。
全局平均池化 - 1¶
版本¶
名称: 全局平均池化 (GitHub)
域:
main
自版本:
1
函数:
False
支持级别:
SupportType.COMMON
形状推断:
True
此版本的运算符自**版本 1**起可用。
摘要¶
全局平均池化使用输入张量 X,并在同一通道中的值上应用平均池化。这等效于内核大小等于输入张量空间维度的平均池化。
输入¶
X (异构) - T
来自先前运算符的输入数据张量;图像情况下的维度为 (N x C x H x W),其中 N 是批大小,C 是通道数,H 和 W 是数据的高度和宽度。对于非图像情况,维度采用 (N x C x D1 x D2 … Dn) 的形式,其中 N 是批大小。
输出¶
Y (异构) - T
跨输入张量池化的输出数据张量。输出张量的秩与输入相同。输出形状的前两个维度与输入相同 (N x C),而其他维度均为 1。
类型约束¶
T in (
tensor(double)
,tensor(float)
,tensor(float16)
)将输入和输出类型约束为浮点张量。