pytorch / 1.8.0 / generated / torch.nn.rrelu.html /

RReLU

class torch.nn.RReLU(lower=0.125, upper=0.3333333333333333, inplace=False) [source]

Applies the randomized leaky rectified liner unit function, element-wise, as described in the paper:

Empirical Evaluation of Rectified Activations in Convolutional Network.

The function is defined as:

RReLU ( x ) = { x if x 0 a x otherwise \text{RReLU}(x) = \begin{cases} x & \text{if } x \geq 0 \\ ax & \text{ otherwise } \end{cases}

where a a is randomly sampled from uniform distribution U ( lower , upper ) \mathcal{U}(\text{lower}, \text{upper}) .

See: https://arxiv.org/pdf/1505.00853.pdf

Parameters
  • lower – lower bound of the uniform distribution. Default: 1 8 \frac{1}{8}
  • upper – upper bound of the uniform distribution. Default: 1 3 \frac{1}{3}
  • inplace – can optionally do the operation in-place. Default: False
Shape:
  • Input: ( N , ) (N, *) where * means, any number of additional dimensions
  • Output: ( N , ) (N, *) , same shape as the input

Examples:

>>> m = nn.RReLU(0.1, 0.3)
>>> input = torch.randn(2)
>>> output = m(input)

© 2019 Torch Contributors
Licensed under the 3-clause BSD License.
https://pytorch.org/docs/1.8.0/generated/torch.nn.RReLU.html