Spec-Zone.ru › PyTorch 1

CELU

class torch.nn.CELU(alpha=1.0, inplace=False) [source]

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

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

Более подробную информацию можно найти в статье Continuously Differentiable Exponential Linear Units .

Параметры:
  • alpha (float) – значение α\alpha для формулы CELU. По умолчанию: 1.0
  • inplace (bool) – необязательно выполнить операцию на месте. По умолчанию: False
Форма:
  • Вход: (∗)(*), где ∗* означает любое количество измерений.
  • Выход: (∗)(*), такая же форма, как и на входе.
../_images/CELU.png

Примеры:

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

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/1.13/generated/torch.nn.CELU.html

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API