Skip to main content

thresholded_relu

Function thresholded_relu 

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

Applies the thresholded rectified linear unit function element-wise.

f(x) =

  • x if x > alpha
  • 0 otherwise

ยงArguments

  • alpha: threshold value (default in ONNX is 1.0).