使用lambd的relu激活函数

Question

嗨，我想用python实现一个lambda函数，它返回x if x>；1和0 otherhwise（relu）：

所以我有短信。比如：

p = [-1,0,2,4,-3,1]

relu_vals = lambda x: x if x>0 else 0 

print(relu_vals(p))

需要注意的是，我希望将lambda的值传递给函数

但它失败了。。。。在

Answer 1

你的程序是对的，但需要一些修改。在

试试这个

>>> p = [-1,0,2,4,-3,1]     
>>> relu_vals = lambda x: x if x>0 else 0     
>>> [relu_vals(i) for i in p] 
[0, 0, 2, 4, 0, 1]

Answer 2

您希望使用map对列表的每个元素应用此函数

list(map(relu_vals, p))

给你

^{pr2}$

另外，如果不打算再次使用lambda函数，最好在map中定义它

print(list(map(lambda x: x if x > 0 else 0, p)))