celu

class torch.ao.nn.quantized.functional.celu(input, scale, zero_point, alpha=1.) [исходный код]

Поэлементно применяет квантованную функцию CELU.

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

Параметры:

• input (Tensor) – квантованный вход
• alpha (float) – значение $\alpha$ для формулы CELU. По умолчанию: 1.0

Тип возвращаемого значения:

Tensor