GlobalAveragePool

GlobalAveragePool - 22

版本

此版本的算子自版本 22 起可用。

摘要

GlobalAveragePool 接收输入张量 X,并在同一通道内的值上应用平均池化。这等效于核大小等于输入张量空间维度的 AveragePool。

输入

  • 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 在 ( tensor(bfloat16), tensor(double), tensor(float), tensor(float16) ) 中

    约束输入和输出类型为浮点张量。

GlobalAveragePool - 1

版本

此版本的算子自版本 1 起可用。

摘要

GlobalAveragePool 接收输入张量 X,并在同一通道内的值上应用平均池化。这等效于核大小等于输入张量空间维度的 AveragePool。

输入

  • 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 在 ( tensor(double), tensor(float), tensor(float16) ) 中

    约束输入和输出类型为浮点张量。