Function shrink 

pub fn shrink<const D: usize>(
    tensor: Tensor<D>,
    lambda: f64,
    bias: f64,
) -> Tensor<D>

Applies the Shrink function element-wise.

shrink(x) = x - bias if x > lambda, x + bias if x < -lambda, 0 otherwise

Arguments

• lambda: the lambda value for the Shrink formulation.
• bias: the bias value for the Shrink formulation.