← 返回函数库
激活函数
ReLU
ReLU(x) = max(0, x)
ReLU 全称 Rectified Linear Unit("修正线性单元")。最简单也最常用的激活:负数置零,正数原样输出。计算便宜、梯度恒等,但有「死神经元」问题。
参数
| 名称 | 类型 | 说明 |
|---|---|---|
| x | float | tensor | 输入 |
返回[0, ∞)
Python 实现
下面是教学用的最小实现,目的是让你看懂原理。生产中请用 PyTorch / TensorFlow 的高性能 op。
python
在计算器里试试
打开底部工具栏的「计算器」,把下面这个表达式贴进表达式输入框,拖动参数滑块看曲线变化:
max(0, x)
ReLU