keras中某一个维做sum操作
2020-06-06 本文已影响0人
YoungDou
今天在keras中实现sum功能，本以为很简单。。。。
K.sum(vec, 1)
结果报错了
AttributeError: 'NoneType' object has no attribute 'inbound_nodes'
使用了K.sum()的方法，它返回的是一个Tensor，而不是一个Layer。
正确的做法应该是使用Lambda层，实现无参数更新的自定义层。
output = Lambda(lambda x: K.sum(x, 1))(vec)
