今天在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)
今天在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)
本文标题：keras中某一个维做sum操作
本文链接：https://www.haomeiwen.com/subject/hokzzhtx.html
网友评论