MaxRoiPool

MaxRoiPool - 22

版本

  • 名称: MaxRoiPool (GitHub)

  • : main

  • 起始版本: 22

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的运算符自 版本 22 起可用。

摘要

ROI 最大池化消耗一个输入张量 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 池化后的 4 维输出张量,形状为 (num_rois, channels, pooled_shape[0], pooled_shape[1])。

类型约束

  • T 位于 ( tensor(bfloat16), tensor(double), tensor(float), tensor(float16) )

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

MaxRoiPool - 1

版本

  • 名称: MaxRoiPool (GitHub)

  • : main

  • 起始版本: 1

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的运算符自 版本 1 起可用。

摘要

ROI 最大池化消耗一个输入张量 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 池化后的 4 维输出张量,形状为 (num_rois, channels, pooled_shape[0], pooled_shape[1])。

类型约束

  • T 位于 ( tensor(double), tensor(float), tensor(float16) )

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