Function dfdx::tensor_ops::leakyrelu

source ·
pub fn leakyrelu<S: Shape, E: Dtype, D: Device<E>, T: Tape<E, D>>(
    lhs: Tensor<S, E, D, T>,
    rhs: E
) -> Tensor<S, E, D, T>
Expand description

Computes prelu, but with a scalar value. max(0, t) + a*min(0, t)