MaxRoiPool
功能
ROI最大池消耗一个输入tensorX和感兴趣区域(ROI),以便在每个ROI上应用最大池,从而产生输出的4-D形状tensor(num_roi, channels, pooled_shape[0], pooled_shape[1])。
输入
两个输入:
x:一个tensor,数据类型:float16。
rois:一个tensor,数据类型:float16。
输出
一个输出:
y:一个tensor,数据类型:float16。
属性
pooled_shape: list of ints。
spatial_scale: float,默认值:1.0。
约束
不支持atc工具参数--precision_mode=must_keep_origin_dtype时float类型输入。
支持的ONNX版本
Opset v8/v9/v10/v11/v12/v13/v14/v15/v16
父主题: 支持ONNX算子清单