python - 如何在python中实现Leaky Relu的导数？

Question

在不使用 Tensorflow 的情况下，如何在 Python 中实现 Leaky ReLU 的导数？

还有比这更好的方法吗？我希望函数返回一个 numpy 数组

def dlrelu(x, alpha=.01):
     # return alpha if x < 0 else 1

     return np.array ([1 if i >= 0 else alpha for i in x])

在此先感谢您的帮助

Answer 1

您使用的方法有效，但严格来说，您正在计算相对于损失或较低层的导数，因此将较低层的值也传递给计算导数 (dl/dx) 可能是明智的。

无论如何，您可以避免使用对大型x. 这是一种方法：

def dlrelu(x, alpha=0.01):
  dx = np.ones_like(x)
  dx[x < 0] = alpha
  return dx

如果您从较低层传递错误，它看起来像这样：

def dlrelu(dl, x, alpha=0.01):
  """ dl and x have same shape. """
  dx = np.ones_like(x)
  dx[x < 0] = alpha
  return dx*dl