keras中某一个维做sum操作

今天在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)
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容