ReLU
-
class torch.nn.ReLU(inplace=False)[source] -
Применяет функцию выпрямленного линейного блока поэлементно:
- Параметры:
-
inplace (bool) – необязательно выполнить операцию на месте. По умолчанию:
False
- Форма:
-
- Вход: , где означает любое количество измерений.
- Выход: , такая же форма, как и вход.
Примеры:
>>> 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