MaxRoiPool

MaxRoiPool - 22

版本

  • 名称: MaxRoiPool (GitHub)

  • 域: main

  • 起始版本: 22

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

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

概要

ROI 最大池化（max pool）使用输入张量 X 和感兴趣区域 (RoIs)，对每个 RoI 应用最大池化，以产生形状为 (num_rois, channels, pooled_shape[0], pooled_shape[1]) 的 4 维输出张量。

属性

  • pooled_shape - INTS (必需)

    ROI 池化输出形状（高度，宽度）。

  • spatial_scale - FLOAT (默认值为 '1.0')

    用于将 ROI 坐标从其输入比例转换为池化时使用的比例的乘性空间比例因子。

输入

  • X (异构) - T

    来自前一个算子的输入数据张量；图像情况下的维度为 (N x C x H x W)，其中 N 是批次大小，C 是通道数，H 和 W 是数据的高度和宽度。

  • rois (异构) - T

    要进行池化的 RoIs（感兴趣区域）。应为一个形状为 (num_rois, 5) 的 2 维张量，形式为 [[batch_id, x1, y1, x2, y2], …]。

输出

  • Y (异构) - T

    RoI 池化输出，一个形状为 (num_rois, channels, pooled_shape[0], pooled_shape[1]) 的 4 维张量。

类型约束

  • T 在 ( tensor(bfloat16), tensor(double), tensor(float), tensor(float16) ) 中

    将输入和输出类型限制为浮点张量。

MaxRoiPool - 1

版本

  • 名称: MaxRoiPool (GitHub)

  • 域: main

  • 起始版本: 1

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

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

概要

ROI 最大池化（max pool）使用输入张量 X 和感兴趣区域 (RoIs)，对每个 RoI 应用最大池化，以产生形状为 (num_rois, channels, pooled_shape[0], pooled_shape[1]) 的 4 维输出张量。

属性

  • pooled_shape - INTS (必需)

    ROI 池化输出形状（高度，宽度）。

  • spatial_scale - FLOAT (默认值为 '1.0')

    用于将 ROI 坐标从其输入比例转换为池化时使用的比例的乘性空间比例因子。

输入

  • X (异构) - T

    来自前一个算子的输入数据张量；图像情况下的维度为 (N x C x H x W)，其中 N 是批次大小，C 是通道数，H 和 W 是数据的高度和宽度。

  • rois (异构) - T

    要进行池化的 RoIs（感兴趣区域）。应为一个形状为 (num_rois, 5) 的 2 维张量，形式为 [[batch_id, x1, y1, x2, y2], …]。

输出

  • Y (异构) - T

    RoI 池化输出，一个形状为 (num_rois, channels, pooled_shape[0], pooled_shape[1]) 的 4 维张量。

类型约束

  • T 在 ( tensor(double), tensor(float), tensor(float16) ) 中

    将输入和输出类型限制为浮点张量。