celu

class torch.ao.nn.quantized.functional.celu(input, scale, zero_point, alpha=1.)

Applies the quantized CELU function element-wise.

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

Parameters

• input (Tensor) – quantized input

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

Return type

Tensor