Skip to main content

leaky_relu

Function leaky_relu 

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

Applies the leaky rectified linear unit function element-wise.

f(x) =

  • x for x >= 0
  • negative_slope * x if x < 0