burn::tensor::activation

Function hard_sigmoid

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

Applies the hard sigmoid function