文档
注册
评分
提单
论坛
小AI

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

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词