CELU

class torch.nn.CELU(alpha=1.0, inplace=False) [исходный код]

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

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

Подробнее см. в статье Непрерывно дифференцируемые экспоненциальные линейные единицы .

Параметры:

• alpha (float) – значение $\alpha$ для формулы CELU. По умолчанию: 1.0
• inplace (bool) – позволяет при необходимости выполнять операцию на месте. По умолчанию: False

Форма:

• Вход: $(*)$, где $*$ означает любое количество измерений.
• Выход: $(*)$, той же формы, что и вход.

Примеры:

>>> m = nn.CELU()
>>> input = torch.randn(2)
>>> output = m(input)

extra_repr() [исходный код]

Возвращает дополнительное представление модуля.

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

str

forward(input) [исходный код]

Выполняет прямой проход.

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

Tensor