MaxRoiPool¶
MaxRoiPool - 22¶
版本¶
域:
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¶
版本¶
域:
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)
) 中将输入和输出类型限制为浮点张量。