ProposalExtract

产 品 支 持 情 况

产 品	是 否 支 持
Ascend 950PR/Ascend 950DT	x
Atlas A3 训 练 系 列 产 品/Atlas A3 推 理 系 列 产 品	x
Atlas A2 训 练 系 列 产 品/Atlas A2 推 理 系 列 产 品	x
Atlas 200I/500 A2 推 理 产 品	x
Atlas 推 理 系 列 产 品AI Core	√
Atlas 推 理 系 列 产 品Vector Core	x
Atlas 训 练 系 列 产 品	√

功 能 说 明

与ProposalConcat功 能 相 反，从Region Proposals内 将 相 应 位 置 的 单 个 元 素 抽 取 后 重 排，每 次 迭 代 处 理16个Region Proposals，抽 取16个 元 素 后 连 续 排 列。

函 数 原 型

Text

template <typename T>
__aicore__ inline void ProposalExtract(const LocalTensor<T>& dst, const LocalTensor<T>& src, const int32_t repeatTime, const int32_t modeNumber)

参 数 说 明

表 1 模 板 参 数 说 明

参 数 名	描 述
T	操 作 数 数 据 类 型。 Atlas 训 练 系 列 产 品，支 持 的 数 据 类 型 为：half Atlas 推 理 系 列 产 品AI Core，支 持 的 数 据 类 型 为：half/float

表 2 参 数 说 明

参 数 名 称	输 入/输 出	含 义
dst	输 出	目 的 操 作 数。 类 型 为LocalTensor，支 持 的TPosition为VECIN/VECCALC/VECOUT。 LocalTensor的 起 始 地 址 需 要32字 节 对 齐。
src	输 入	源 操 作 数。 类 型 为LocalTensor，支 持 的TPosition为VECIN/VECCALC/VECOUT。 LocalTensor的 起 始 地 址 需 要32字 节 对 齐。 源 操 作 数 的 数 据 类 型 需 要 与 目 的 操 作 数 保 持 一 致。
repeatTime	输 入	重 复 迭 代 次 数，int32_t类 型，每 次 迭 代 完 成16个Region Proposals的 元 素 抽 取 并 排 布 到16个 元 素 里，下 次 迭 代 跳 至 相 邻 的 下 一 组16个Region Proposals和 下 一 组16个 元 素。取 值 范 围：repeatTime∈[0,255]。
modeNumber	输 入	抽 取 位 置 参 数，取 值 范 围：modeNumber∈[0, 5]，int32_t类 型，仅 限 于 以 下 配 置： 0 – 从x1抽 取 1 – 从y1抽 取 2 – 从x2抽 取 3 – 从y2抽 取 4 – 从score抽 取 5 – 从label抽 取

返 回 值 说 明

无

约 束 说 明

• 用 户 需 保 证src中 存 储 的proposal数 目 大 于 等 于 实 际 所 需 数 目，否 则 会 存 在tensor越 界 错 误。
• 用 户 需 保 证dst中 存 储 的 元 素 大 于 等 于 实 际 所 需 数 目，否 则 会 存 在tensor越 界 错 误。
• 操 作 数 地 址 对 齐 要 求 请 参 见通 用 地 址 对 齐 约 束。

调 用 示 例

详 细 示 例 请 参 考RegionProposalSort样 例。

• 接 口 使 用 样 例

  Text

  // repeatTime = 2, modeNumber = 4, 把32个Region Proposal中 的score域 元 素 抽 取 出 来 排 列 成32个 连 续 元 素
  AscendC::ProposalExtract(dstLocal, srcLocal, 2, 4);
  

  Text

  示 例 结 果 
  输 入 数 据(src_gm):
  因 为moodel=4，第 一 个 元 素33.3的 起 始 位 置 是4。每 个Region Proposal占 用 连 续8个half/float类 型 的 元 素。这 里 使 用 的 类 型 是half。后 续 被 抽 取 的 每 个 元 素 间 隔8个 元 素。repeat为2，每 次 迭 代 完 成16个 元 素，共 计32个 元 素
  [ 0.      0.      0.      0. 
    33.3    0.      0.      0.      0.      0.      0.      0.
    67.56   0.      0.      0.      0.      0.      0.      0.
    68.5    0.      0.      0.      0.      0.      0.      0.
    -11.914 0.      0.      0.      0.      0.      0.      0.
    25.19   0.      0.      0.      0.      0.      0.      0.
    -72.8   0.      0.      0.      0.      0.      0.      0.
    11.79   0.      0.      0.      0.      0.      0.      0.
    -49.47  0.      0.      0.      0.      0.      0.      0.
    49.44   0.      0.      0.      0.      0.      0.      0.
    84.4    0.      0.      0.      0.      0.      0.      0.
    -14.36  0.      0.      0.      0.      0.      0.      0.
    45.97   0.      0.      0.      0.      0.      0.      0.
    52.47   0.      0.      0.      0.      0.      0.      0.
    -5.387  0.      0.      0.      0.      0.      0.      0.
    -13.12  0.      0.      0.      0.      0.      0.      0.
    -88.9   0.      0.      0.      0.      0.      0.      0.
    54.     0.      0.      0.      0.      0.      0.      0.
    -51.62  0.      0.      0.      0.      0.      0.      0.
   -20.67   0.      0.      0.      0.      0.      0.      0.
   59.56    0.      0.      0.      0.      0.      0.      0.
   35.72    0.      0.      0.      0.      0.      0.      0.
   -6.12    0.      0.      0.      0.      0.      0.      0.
   -39.4    0.      0.      0.      0.      0.      0.      0.
   -11.46   0.      0.      0.      0.      0.      0.      0.
   -7.066   0.      0.      0.      0.      0.      0.      0.
   30.23    0.      0.      0.      0.      0.      0.      0.
   -11.18   0.      0.      0.      0.      0.      0.      0.
   -35.84   0.      0.      0.      0.      0.      0.      0.
   -40.88   0.      0.      0.      0.      0.      0.      0.
   60.9     0.      0.      0.      0.      0.      0.      0.
   -73.3    0.      0.      0.      0.      0.      0.      0.
   38.47    0.      0.      0. 
   ]
  输 出 数 据(dst_gm):
  [ 33.3    67.56   68.5   -11.914  25.19  -72.8    11.79  -49.47   49.44
    84.4   -14.36   45.97   52.47   -5.387 -13.12  -88.9    54.    -51.62
   -20.67   59.56   35.72   -6.12  -39.4   -11.46   -7.066  30.23  -11.18
   -35.84  -40.88   60.9   -73.3    38.47 ]