请问如何在keras中给某一层添加门限

比如在下列代码中,output输出的结果是一个N维张量,每个神经元输出一个0~1之间的数

output = Dense(N, activation='sigmoid')(x)

我希望对整个output层添加一个门限,来使整个层的输出变成只有0或1的张量。具体操作:令所有大于0.5的值为1,小于0.5的值为0 直观上应该用Lambda层来实现,但是具体应当如何操作呢?

不知道你这个问题是否已经解决, 如果还没有解决的话:

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^