- class torch.nn.ELU(alpha=1.0, inplace=False)¶
Applies the Exponential Linear Unit (ELU) function, element-wise, as described in the paper: Fast and Accurate Deep Network Learning by Exponential Linear Units (ELUs).
ELU is defined as:
Input: , where means any number of dimensions.
Output: , same shape as the input.
>>> m = nn.ELU() >>> input = torch.randn(2) >>> output = m(input)