Spec-Zone.ru › PyTorch 2

LeakyReLU

class torch.nn.LeakyReLU(negative_slope=0.01, inplace=False) [source]

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

LeakyReLU(x)=max⁡(0,x)+negative_slope∗min⁡(0,x)\text{LeakyReLU}(x) = \max(0, x) + \text{negative\_slope} * \min(0, x)

или

LeakyReLU(x)={x, если x≥0negative_slope×x, в противном случае \text{LeakyReLU}(x) = \begin{cases} x, & \text{ if } x \geq 0 \\ \text{negative\_slope} \times x, & \text{ otherwise } \end{cases}
Параметры
  • negative_slope (float) – Управляет углом отрицательного наклона (который используется для отрицательных значений ввода). Значение по умолчанию: 1e-2
  • inplace (bool) – необязательно выполняет операцию на месте. Значение по умолчанию: False
Форма:
  • Вход: (∗)(*) где * означает любое количество дополнительных измерений
  • Выход: (∗)(*), форма такая же, как у входных данных
../_images/LeakyReLU.png

Примеры:

>>> m = nn.LeakyReLU(0.1)
>>> 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/2.1/generated/torch.nn.LeakyReLU.html

Spec-Zone.ru

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