RpSort16

产 品 支 持 情 况

产 品	是 否 支 持
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 训 练 系 列 产 品	√

功 能 说 明

根 据Region Proposals中 的score域 对 其 进 行 排 序（score大 的 排 前 面），每 次 排16个Region Proposals。

函 数 原 型

Text

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

参 数 说 明

表 1 模 板 参 数 说 明

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

表 2 参 数 说 明

参 数 名 称	输 入/输 出	含 义
dst	输 出	目 的 操 作 数，存 储 经 过 排 序 后 的Region Proposals。 类 型 为LocalTensor，支 持 的TPosition为VECIN/VECCALC/VECOUT。 LocalTensor的 起 始 地 址 需 要32字 节 对 齐。
src	输 入	源 操 作 数，存 储 未 经 过 排 序 的Region Proposals。 类 型 为LocalTensor，支 持 的TPosition为VECIN/VECCALC/VECOUT。 LocalTensor的 起 始 地 址 需 要32字 节 对 齐。
repeatTime	输 入	重 复 迭 代 次 数，int32_t类 型，每 次 排16个Region Proposals。取 值 范 围：repeatTime∈[0,255]。

约 束 说 明

• 用 户 需 保 证src和dst中 存 储 的Region Proposal数 目 大 于 实 际 所 需 数 据，否 则 会 存 在tensor越 界 错 误。
• 当 存 在proposal[i]与proposal[j]的score值 相 同 时，如 果i>j，则proposal[j]将 首 先 被 选 出 来，排 在 前 面。
• 操 作 数 地 址 对 齐 要 求 请 参 见通 用 地 址 对 齐 约 束。

调 用 示 例

详 细 示 例 请 参 考RegionProposalSort样 例。

• 接 口 使 用 样 例

  Text

  // ProposalConcat将 连 续 元 素 合 入Region Proposal内 对 应 位 置
  // repeatTime = 2, 对2个Region Proposal进 行 排 序，model=4起 始 位 置 为4
  AscendC::ProposalConcat(dstLocal, srcLocal, 2, 4);
  AscendC::RpSort16(dstLocal, dstLocal, 2);
  

  Text

  示 例 结 果
  输 入 数 据srcLocal:
  [ -1.624 -42.3   -54.12   91.25  -99.4    36.72   67.44  -66.3   -52.53
     3.377 -62.47  -15.85  -31.47    3.143  58.47  -83.75 21.58   63.47    
     7.234  35.16  -39.72   37.8    73.06  -98.7    44.1 -77.2    67.2    
     19.62  -87.9   -14.875  15.86  -77.75]
  经 过ProposalConcat后 的dstLocal数 据，repeat=2计 算32个 元 素，model=4起 始 位 置 为4
  [  
   0.        0.      0.      0.
  -1.624     0.      0.      0.      0.      0.      0.      0.
  -42.3      0.      0.      0.      0.      0.      0.      0.
  -54.12     0.      0.      0.      0.      0.      0.      0.
  91.25      0.      0.      0.      0.      0.      0.      0.
  -99.4      0.      0.      0.      0.      0.      0.      0.
  36.72      0.      0.      0.      0.      0.      0.      0.
  67.44      0.      0.      0.      0.      0.      0.      0.
  -66.3      0.      0.      0.      0.      0.      0.      0.
  -52.53     0.      0.      0.      0.      0.      0.      0.
  3.377      0.      0.      0.      0.      0.      0.      0.
  -62.47     0.      0.      0.      0.      0.      0.      0.
  -15.85     0.      0.      0.      0.      0.      0.      0.
  -31.47     0.      0.      0.      0.      0.      0.      0.
  3.143      0.      0.      0.      0.      0.      0.      0.
  58.47      0.      0.      0.      0.      0.      0.      0.
  -83.75     0.      0.      0.      0.      0.      0.      0.
  21.58      0.      0.      0.      0.      0.      0.      0.
  63.47      0.      0.      0.      0.      0.      0.      0.
  7.234      0.      0.      0.      0.      0.      0.      0.
  35.16      0.      0.      0.      0.      0.      0.      0.
  -39.72     0.      0.      0.      0.      0.      0.      0.
  37.8       0.      0.      0.      0.      0.      0.      0.
  73.06      0.      0.      0.      0.      0.      0.      0.
  -98.7      0.      0.      0.      0.      0.      0.      0.
  44.1       0.      0.      0.      0.      0.      0.      0.
  -77.2      0.      0.      0.      0.      0.      0.      0.
  67.2       0.      0.      0.      0.      0.      0.      0.
  19.62      0.      0.      0.      0.      0.      0.      0.
  -87.9      0.      0.      0.      0.      0.      0.      0.
  -14.875    0.      0.      0.      0.      0.      0.      0.
  15.86      0.      0.      0.      0.      0.      0.      0.
  -77.75     0.      0.      0.     
  ]
  输 出 数 据(dst_gm):
   [
   0.      0.      0.      0.
   91.25   0.      0.      0.      0.      0.      0.      0.
   67.44   0.      0.      0.      0.      0.      0.      0.
   58.47   0.      0.      0.      0.      0.      0.      0.
   36.72   0.      0.      0.      0.      0.      0.      0.
   3.377   0.      0.      0.      0.      0.      0.      0.
   3.143   0.      0.      0.      0.      0.      0.      0.
   -1.624  0.      0.      0.      0.      0.      0.      0.
   -15.85  0.      0.      0.      0.      0.      0.      0.
   -31.47  0.      0.      0.      0.      0.      0.      0.
   -42.3   0.      0.      0.      0.      0.      0.      0.
   -52.53  0.      0.      0.      0.      0.      0.      0.
   -54.12  0.      0.      0.      0.      0.      0.      0.
   -62.47  0.      0.      0.      0.      0.      0.      0.
   -66.3   0.      0.      0.      0.      0.      0.      0
   -83.75  0.      0.      0.      0.      0.      0.      0.
   -99.4   0.      0.      0.      0.      0.      0.      0.
   73.06   0.      0.      0.      0.      0.      0.      0.
   67.2    0.      0.      0.      0.      0.      0.      0.
   63.47   0.      0.      0.      0.      0.      0.      0.
   44.1    0.      0.      0.      0.      0.      0.      0.
   37.8    0.      0.      0.      0.      0.      0.      0.
   35.16   0.      0.      0.      0.      0.      0.      0.
   21.58   0.      0.      0.      0.      0.      0.      0.
   19.62   0.      0.      0.      0.      0.      0.      0.
   15.86   0.      0.      0.      0.      0.      0.      0.
   7.234   0.      0.      0.      0.      0.      0.      0.
   -14.875 0.      0.      0.      0.      0.      0.      0.
   -39.72  0.      0.      0.      0.      0.      0.      0.
   -77.2   0.      0.      0.      0.      0.      0.      0.
   -77.75  0.      0.      0.      0.      0.      0.      0.
   -87.9   0.      0.      0.      0.      0.      0.      0.
   -98.7   0.      0.      0.
   ]