使用NPU API进行PSROIPool。
[object Object]
- pooled_height (
[object Object]) - 池化高度。 - pooled_width (
[object Object]) - 池化宽度。 - spatial_scale (
[object Object]) - 按此参数值缩放输入框。 - group_size (
[object Object]) - 编码位置敏感分数图的组数。 - output_dim (
[object Object]) - 输出通道数。
[object Object]
shape为(k, 5)和(k, 1)的张量。标签以0为基础。
仅实现了pooled_height == pooled_width == group_size。
[object Object]