celu

Function celu 

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

Applies the Continuously Differentiable Exponential Linear Unit function element-wise.

celu(x) = max(0, x) + min(0, alpha * (exp(x / alpha) - 1))

See also CELU

ยงArguments

  • alpha: scaling parameter for the negative part.