解决keras加入lambda层时shape的问题
作者:nuaa_bo 时间:2023-05-19 17:47:29
使用keras时,加入keras的lambda层以实现自己定义的操作。但是,发现操作结果的shape信息有问题。
我的后端是theano,使用了sum操作。
比如输入时,shape为(32,28,28),其中32为batch大小。
此时对应的ndim应该等于3。
但是,lambda处理后结果显示_keras_shape为(32,28,28),而ndim却是2。
这导致后边各项操作都会出现问题。
此处sum函数加入参数keepdims=True即可。
此注意keras中的各种层几乎都不用去理会batch的大小,系统会自动在shape中加入None占位,所以很多参数也不用加入batch的大小。但是进行sum等操作时,选择按照哪个axis进行操作,要考虑batch的存在。
补充知识:keras Merge or merge
在使用keras merge层时,发现有Merge 、merge两种:
from keras.layers import Merge
from keras.layers import merge
使用第一种是报错
“TensorVariable object has no attribute 'get_output_shape_at' ”
使用第二种小写即可。
来源:https://blog.csdn.net/u014749291/article/details/61927162
标签:keras,lambda,shape


猜你喜欢
php连接微软MSSQL(sql server)完全攻略
2023-07-16 17:56:43

MySql中如何使用 explain 查询 SQL 的执行计划
2024-01-15 10:25:01
pandas检查和填充缺失值的N种方法总结
2021-03-27 03:25:21

删除python pandas.DataFrame 的多重index实例
2022-11-10 17:25:12
基于Python闭包及其作用域详解
2023-11-07 07:22:02

vuejs实现下拉框菜单选择
2023-09-23 08:49:54

微信小程序封装多张图片上传api代码实例
2024-04-25 13:13:31
JS循环中正确使用async、await的姿势分享
2024-05-25 15:18:56

python爬虫lxml库解析xpath网页过程示例
2021-10-06 20:01:54

Pycharm学习教程(5) Python快捷键相关设置
2021-08-08 18:56:44

Python实现GIF动图以及视频卡通化详解
2022-06-28 03:27:34

一文教会你用python连接并简单操作SQLserver数据库
2024-01-17 14:11:34

Python编程给numpy矩阵添加一列方法示例
2023-08-29 07:22:30
OpenCV imread读取图片失败的问题及解决
2021-10-20 06:44:00

javascript应用:Iframe自适应其加载的内容高度
2024-02-25 13:46:07
Python中如何使用Matplotlib库绘制图形
2021-09-27 11:59:04

详解python中文编码问题
2023-10-17 12:09:08

python递归函数求n的阶乘,优缺点及递归次数设置方式
2022-12-08 16:17:08

浅谈python输出列表元素的所有排列形式
2023-02-10 11:09:39
如何把一长串数字分位显示?
2009-11-06 14:01:00