使用 lambda 的 relu 激活函数

Question

嗨，我想在 python 中实现一个 lambda 函数，如果 x> 1 和 0 则返回 x (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

您想使用 map 将此函数应用于列表的每个元素

list(map(relu_vals, p))

给你

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

此外，如果您不打算再次使用它，最好在 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]

relu activation function using lambda