celu

class torch.nn.quantized.functional.celu(input, scale, zero_point, alpha=1.0) → Tensor

Applies the quantized CELU function element-wise.

$$\text{CELU}(x) = \max(0,x) + \min(0, \alpha * (\exp(x / \alpha) - 1))$$

Parameters

• input – quantized input

• alpha – the $\alpha$ value for the CELU formulation. Default: 1.0