MaxRoiPool¶
MaxRoiPool - 22¶
版本¶
域:
main自版本:
22功能:
False支持级别:
SupportType.COMMON形状推断:
True
此版本的运算符自版本 22起可用。
摘要¶
ROI 最大池化消耗输入张量 X 和感兴趣区域 (RoI) 来对每个 RoI 应用最大池化，以生成形状为 (num_rois, channels, pooled_shape[0], pooled_shape[1]) 的输出 4-D 张量。
属性¶
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
要池化的 RoI (感兴趣区域)。应为形状为 (num_rois, 5) 的 2-D 张量，表示为 [[batch_id, x1, y1, x2, y2], …]。
输出¶
Y (异构) - T
形状为 (num_rois, channels, pooled_shape[0], pooled_shape[1]) 的 RoI 池化输出 4-D 张量。
类型约束¶
T in (
tensor(bfloat16),tensor(double),tensor(float),tensor(float16))将输入和输出类型约束为浮点张量。
MaxRoiPool - 1¶
版本¶
域:
main自版本:
1功能:
False支持级别:
SupportType.COMMON形状推断:
True
此版本的运算符自版本 1起可用。
摘要¶
ROI 最大池化消耗输入张量 X 和感兴趣区域 (RoI) 来对每个 RoI 应用最大池化，以生成形状为 (num_rois, channels, pooled_shape[0], pooled_shape[1]) 的输出 4-D 张量。
属性¶
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
要池化的 RoI (感兴趣区域)。应为形状为 (num_rois, 5) 的 2-D 张量，表示为 [[batch_id, x1, y1, x2, y2], …]。
输出¶
Y (异构) - T
形状为 (num_rois, channels, pooled_shape[0], pooled_shape[1]) 的 RoI 池化输出 4-D 张量。
类型约束¶
T in (
tensor(double),tensor(float),tensor(float16))将输入和输出类型约束为浮点张量。