Spec-Zone.ru › PyTorch 1

ReLU

class torch.nn.ReLU(inplace=False) [source]

Применяет функцию выпрямленного линейного блока поэлементно:

ReLU(x)=(x)+=max⁡(0,x)\text{ReLU}(x) = (x)^+ = \max(0, x)

Параметры:

inplace (bool) – необязательно выполнить операцию на месте. По умолчанию: False

Форма:
  • Вход: (∗)(*), где ∗* означает любое количество измерений.
  • Выход: (∗)(*), такая же форма, как и вход.
../_images/ReLU.png

Примеры:

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


An implementation of CReLU - https://arxiv.org/abs/1603.05201

  >>> m = nn.ReLU()
  >>> input = torch.randn(2).unsqueeze(0)
  >>> output = torch.cat((m(input),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.ReLU.html

Spec-Zone.ru

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