MaxRoiPool

MaxRoiPool - 22

版本

  • 名稱: MaxRoiPool (GitHub)

  • 網域: main

  • since_version: 22

  • 函數: False

  • 支援層級: SupportType.COMMON

  • 形狀推斷: True

此運算子版本自版本 22 起可用。

摘要

ROI 最大池化會消耗輸入張量 X 和感興趣區域 (RoI)，以在每個 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

    要池化的 RoI (感興趣區域)。應該是形狀為 (num_rois, 5) 的 2 維張量，格式為 [[batch_id, x1, y1, x2, y2], …]。

輸出

  • Y (異質) - T

    ROI 池化輸出 4 維張量，形狀為 (num_rois, channels, pooled_shape[0], pooled_shape[1])。

類型限制

  • T in ( tensor(bfloat16), tensor(double), tensor(float), tensor(float16) )

    將輸入和輸出類型限制為浮點數張量。

MaxRoiPool - 1

版本

  • 名稱: MaxRoiPool (GitHub)

  • 網域: main

  • since_version: 1

  • 函數: False

  • 支援層級: SupportType.COMMON

  • 形狀推斷: True

此運算子版本自版本 1 起可用。

摘要

ROI 最大池化會消耗輸入張量 X 和感興趣區域 (RoI)，以在每個 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

    要池化的 RoI (感興趣區域)。應該是形狀為 (num_rois, 5) 的 2 維張量，格式為 [[batch_id, x1, y1, x2, y2], …]。

輸出

  • Y (異質) - T

    ROI 池化輸出 4 維張量，形狀為 (num_rois, channels, pooled_shape[0], pooled_shape[1])。

類型限制

  • T in ( tensor(double), tensor(float), tensor(float16) )

    將輸入和輸出類型限制為浮點數張量。