Function burn::tensor::activation::leaky_relu

pub fn leaky_relu<const D: usize, B>(
    tensor: Tensor<B, D>,
    negative_slope: f64,
) -> Tensor<B, D>
where
    B: Backend,
Expand description

Applies the leaky rectified linear unit function.

f(x) = negative_slope * x for x < 0, f(x) = x for x >= 0