我試圖實現一個函數,它計算矩陣中每個元素的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) –