美文网首页
keras中某一个维做sum操作

keras中某一个维做sum操作

作者: YoungDou | 来源:发表于2020-06-06 08:12 被阅读0次

今天在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