我试图实现一个函数,它计算矩阵中每个元素的Relu导数,然后将结果返回到矩阵中。我正在使用Python和Numpy。在python numpy中实现Relu派生
基于其它交叉验证帖,x的RELU衍生物是 1当x> 0时,0当x < 0,未定义或0当x == 0
目前,我有以下代码,以便远:
def reluDerivative(self, x):
return np.array([self.reluDerivativeSingleElement(xi) for xi in x])
def reluDerivativeSingleElement(self, xi):
if xi > 0:
return 1
elif xi <= 0:
return 0
不幸的是,xi是一个数组,因为x是一个矩阵。 reluDerivativeSingleElement函数不适用于数组。所以我想知道是否有一种方法可以使用numpy将矩阵中的值映射到另一个矩阵,就像numpy中的exp函数一样?
非常感谢。
[np.heaviside(https://docs.scipy.org/doc/numpy/reference/generated/numpy.heaviside.html) –