使用lambda的relu激活函数

Question

嗨，我想在python中实现一个lambda函数，它给我返回x如果x> 1和0 (relu)：

所以我有smth。像这样：

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

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

print(relu_vals(p))

请注意，我希望将lambda的值传递给一个函数。

但失败了..。

Answer 1

您希望使用地图对列表中的每个元素应用此函数。

list(map(relu_vals, p))

给你

[0, 0, 2, 4, 0, 1]

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

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

Answer 2

你的程序是正确的，但需要一些修改。

尝尝这个,

>>> 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]