keras 获取某层的输入/输出 tensor 尺寸操作


Posted in Python onJune 10, 2020

获取单输入尺寸,该层只被使用了一次。

import keras
from keras.layers import Input, LSTM, Dense, Conv2D
from keras.models import Model
a = Input(shape=(32, 32, 3))
b = Input(shape=(64, 64, 3))
 
conv = Conv2D(16, (3, 3), padding='same')
conved_a = conv(a)
 
# 到目前为止只有一个输入,以下可行:
assert conv.input_shape == (None, 32, 32, 3)

如果该层被使用了两次

import keras
from keras.layers import Input, LSTM, Dense, Conv2D
from keras.models import Model
a = Input(shape=(32, 32, 3))
b = Input(shape=(64, 64, 3))
 
conv = Conv2D(16, (3, 3), padding='same')
conved_a = conv(a)
 
# 到目前为止只有一个输入,以下可行:
assert conv.input_shape == (None, 32, 32, 3)
 
conved_b = conv(b)
# 现在 `.input_shape` 属性不可行,但是这样可以:
assert conv.get_input_shape_at(0) == (None, 32, 32, 3)
assert conv.get_input_shape_at(1) == (None, 64, 64, 3)

如果是输出,只需要改成output就好:

import keras
from keras.layers import Input, LSTM, Dense, Conv2D
from keras.models import Model
a = Input(shape=(32, 32, 3))
b = Input(shape=(64, 64, 3))
 
conv = Conv2D(16, (3, 3), padding='same')
conved_a = conv(a)
 
# 到目前为止只有一个输入,以下可行:
assert conv.input_shape == (None, 32, 32, 3)
 
conved_b = conv(b)
# 就改了output,当然尺寸我也改了
assert conv.get_output_shape_at(0) == (None, 32, 32, 16)
assert conv.get_output_shape_at(1) == (None, 64, 64, 16)

补充知识:keras中获取shape的正确方法

在keras的网络中,如果用layer_name.shape的方式获取shape信息将会返还tensorflow.python.framework.tensor_shape.TensorShape其中包含的是tensorflow.python.framework.tensor_shape.Dimension

正确的方式是使用

import keras.backend as K
K.int_shape(laye_name)

以上这篇keras 获取某层的输入/输出 tensor 尺寸操作就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
详解JavaScript编程中的window与window.screen对象
Oct 26 Python
Python基于sftp及rsa密匙实现远程拷贝文件的方法
Sep 21 Python
从请求到响应过程中django都做了哪些处理
Aug 01 Python
使用pandas读取文件的实现
Jul 31 Python
pyhton中__pycache__文件夹的产生与作用详解
Nov 24 Python
Python 将json序列化后的字符串转换成字典(推荐)
Jan 06 Python
解决Python命令行下退格,删除,方向键乱码(亲测有效)
Jan 16 Python
PyCharm无法引用自身项目解决方式
Feb 12 Python
Python基于requests库爬取网站信息
Mar 02 Python
python中shell执行知识点
May 06 Python
使用matlab 判断两个矩阵是否相等的实例
May 11 Python
Python轻量级web框架bottle使用方法解析
Jun 13 Python
Python 字典中的所有方法及用法
Jun 10 #Python
在keras 中获取张量 tensor 的维度大小实例
Jun 10 #Python
Keras—embedding嵌入层的用法详解
Jun 10 #Python
Keras框架中的epoch、bacth、batch size、iteration使用介绍
Jun 10 #Python
Python3.9 beta2版本发布了,看看这7个新的PEP都是什么
Jun 10 #Python
JAVA及PYTHON质数计算代码对比解析
Jun 10 #Python
keras 使用Lambda 快速新建层 添加多个参数操作
Jun 10 #Python
You might like
php编写的简单页面跳转功能实现代码
2013/11/27 PHP
javascript中的prototype属性使用说明(函数功能扩展)
2010/08/16 Javascript
JQuery实现表格中相同单元格合并示例代码
2013/06/26 Javascript
textarea不能通过maxlength属性来限制字数的解决方法
2014/09/01 Javascript
jQuery中addClass()方法用法实例
2015/01/05 Javascript
javascirpt实现2个iframe之间传值的方法
2016/06/30 Javascript
原生js实现键盘控制div移动且解决停顿问题
2016/12/05 Javascript
jQuery实现立体式数字滚动条增加效果
2016/12/21 Javascript
vue.js开发环境安装教程
2017/03/17 Javascript
node.js基于fs模块对系统文件及目录进行读写操作的方法详解
2017/11/10 Javascript
vuex实现的简单购物车功能示例
2019/02/13 Javascript
原生js+canvas实现贪吃蛇效果
2020/08/02 Javascript
如何通过Proxy实现JSBridge模块化封装
2020/10/22 Javascript
python根据文件大小打log日志
2014/10/09 Python
Python求导数的方法
2015/05/09 Python
基于Python3 逗号代码 和 字符图网格(详谈)
2017/06/22 Python
python+pillow绘制矩阵盖尔圆简单实例
2018/01/16 Python
python实现音乐下载的统计
2018/06/20 Python
Python3实现从排序数组中删除重复项算法分析
2019/04/03 Python
python买卖股票的最佳时机(基于贪心/蛮力算法)
2019/07/05 Python
python实现网站用户名密码自动登录功能
2019/08/09 Python
使用Pycharm(Python工具)新建项目及创建Python文件的教程
2020/04/26 Python
python中return如何写
2020/06/18 Python
CSS3 完美实现圆角效果
2009/07/13 HTML / CSS
html5 touch事件实现页面上下滑动效果【附代码】
2016/03/10 HTML / CSS
js实现弹框效果
2021/03/24 Javascript
中英文自我评价常用句型
2013/12/19 职场文书
四年级数学教学反思
2014/02/02 职场文书
《秋姑娘的信》教学反思
2014/02/28 职场文书
教师年度考核评语
2014/04/28 职场文书
安全承诺书格式
2014/05/21 职场文书
阅兵口号
2014/06/19 职场文书
大学生赌博检讨书
2014/09/22 职场文书
2015新学期校长寄语(3篇)
2015/03/25 职场文书
2015年施工员工作总结范文
2015/04/20 职场文书
python pyhs2 的安装操作
2021/04/07 Python