如何在numpy中实现这个激活函数?

2024-04-29 10:05:33 发布

您现在位置:Python中文网/ 问答频道 /正文

如何使用numpy实现:

enter image description here

它的导数f'(x)?我试着:

    def func (x,y):
        if x.all() <= 0:
           y = beta (np.exp(x)-1)
           return y
        elif x>0:
            y = x
            return y

但是他们不起作用。注意,x和{}是数组。在


Tags: numpyreturnifdefnp数组allbeta
1条回答
网友
1楼 · 发布于 2024-04-29 10:05:33

怎么样:

def func (x, beta):
    y = np.empty_like(x)
    mask = x <= 0
    y[mask] = beta * (np.exp(x[mask])-1)
    y[~mask] = x[~mask]
   return y

mask包含<;=0的元素的索引,因此可以将这两种情况分开。在

相关问题 更多 >