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

Question

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

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

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

Answer 1

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

• 给你找了一篇非常好的博客，你可以看看是否有帮助，链接：Keras加载预训练模型

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