asc_leakyrelu

产 品 支 持 情 况

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

功 能 说 明

按 元 素 执 行Leaky ReLU（Leaky Rectified Linear Unit）操 作。Leaky ReLU带 泄 露 性 整 流 函 数 是 一 种 人 工 神 经 网 络 中 常 用 的 激 活 函 数，其 数 学 表 达 式 为：

$$ f(x_i) = \begin{cases} x_i & \text{if } x_i ≥ \text{ }0 \ \ ax_i & \text{if } x_i < \text{ }0 \ \end{cases} $$

函 数 原 型

C++

__simd_callee__ inline void asc_leakyrelu(vector_float& dst, vector_float src, float value, vector_bool mask)
__simd_callee__ inline void asc_leakyrelu(vector_half& dst, vector_half src, half value, vector_bool mask)

参 数 说 明

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

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

返 回 值 说 明

无

流 水 类 型

PIPE_V

约 束 说 明

无

调 用 示 例

C++

vector_half dst;
vector_half src;
half value = 0.1;
vector_bool mask = asc_create_mask_b16(PAT_ALL);
asc_loadalign(src, src_addr); // src_addr是 外 部 输 入 的UB内 存 空 间 地 址。
asc_leakyrelu(dst, src, value, mask);