ELU¶
- class torch.nn.ELU(alpha=1.0, inplace=False)[source][source]¶
应用指数线性单元(ELU)函数,逐元素。
方法见论文:通过指数线性单元(ELUs)实现快速且精确的深度网络学习。
ELU 定义为:
- 参数:
alpha(浮点数)- ELU 公式的 值。默认:1.0
inplace(布尔值)- 可选地进行原地操作。默认:
False
- 形状:
输入: ,其中 表示任意数量的维度。
输出: ,与输入具有相同的形状。
示例:
>>> m = nn.ELU() >>> input = torch.randn(2) >>> output = m(input)