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 \ast (\exp (x/\alpha )-1))$$

Parameters

• input (Tensor) – quantized input

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

Return type

Tensor