On this page
LeakyReLU
class torch.nn.LeakyReLU(negative_slope=0.01, inplace=False)
[source]-
Applies the element-wise function:
or
- Parameters
- Shape:
-
- Input:
where
*
means, any number of additional dimensions - Output: , same shape as the input
- Input:
where
Examples:
>>> 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