python - 使用 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]