asc_prelu

产 品 支 持 情 况

产 品	是 否 支 持
Ascend 950PR/Ascend 950DT	√

功 能 说 明

源 操 作 数src0大 于0的 情 况 下 直 接 将src0写 入 目 的 操 作 数dst，否 则 将src0 * src1的 结 果 写 入dst。计 算 公 式 如 下：

$$ dst_i = (src0_i > 0) ? src0_i : src0_i * src1_i $$

函 数 原 型

C++

__simd_callee__ inline void asc_prelu(vector_float& dst, vector_float src0, vector_float src1, vector_bool mask)
__simd_callee__ inline void asc_prelu(vector_half& dst, vector_half src0, vector_half src1, vector_bool mask)

参 数 说 明

参 数 名	输 入/输 出	描 述
dst	输 出	目 的 操 作 数（矢 量 数 据 寄 存 器）。
src0	输 入	源 操 作 数（矢 量 数 据 寄 存 器）。
src1	输 入	源 操 作 数（矢 量 数 据 寄 存 器）。
mask	输 入	源 操 作 数 掩 码（掩 码 寄 存 器），用 于 指 示 在 计 算 过 程 中 哪 些 元 素 参 与 计 算。对 应 位 置 为1时 参 与 计 算，为0时 不 参 与 计 算。mask未 筛 选 的 元 素 在 输 出 中 置 零。

矢 量 数 据 寄 存 器 和 掩 码 寄 存 器 的 详 细 说 明 请 参 见reg数 据 类 型 定 义。

返 回 值 说 明

无

流 水 类 型

PIPE_V

约 束 说 明

无

调 用 示 例

C++

vector_half dst;
vector_half src0;
vector_half src1;
vector_bool mask = asc_create_mask_b16(PAT_ALL);
asc_loadalign(src0, src0_addr); // src0_addr是 外 部 输 入 的UB内 存 空 间 地 址。
asc_loadalign(src1, src1_addr); // src1_addr是 外 部 输 入 的UB内 存 空 间 地 址。
asc_prelu(dst, src0, src1, mask);