vlrelu
功能说明
Leaky ReLU 接口,如果元素大于或等于0,则保持原值不变,如果元素小于0,则原值乘以 src1 定义的标量值,计算公式如下:
each_element_of([dst]) = each_element_of([src0]) < 0 ? each_element_of([src0]) * src1 : each_element_of([src0])
以 block(32Byte)为单位完成计算,一次完成8个 block 的计算。
该接口支持通过MASK控制哪些元素参与计算。
接口原型
// 相同接口的不同原型区别在于源地址和目的地址的数据类型不同。 void vlrelu(__ubuf__ half *dst, __ubuf__ half *src0, half src1, uint8_t repeat, uint16_t dstBlockStride, uint16_t src0BlockStride, uint16_t dstRepeatStride, uint16_t src0RepeatStride); void vlrelu(__ubuf__ float *dst, __ubuf__ float *src0, float src1, uint8_t repeat, uint16_t dstBlockStride, uint16_t src0BlockStride, uint16_t dstRepeatStride, uint16_t src0RepeatStride);
参数说明
参数含义见 表2 双目运算参数说明。
流水类型
PIPE_V
父主题: 双目运算