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 相关文章推荐
python在多玩图片上下载妹子图的实现代码
Aug 13 Python
pycharm 使用心得(五)断点调试
Jun 06 Python
OpenCV实现人脸识别
Apr 07 Python
Python 的类、继承和多态详解
Jul 16 Python
python实现微信接口(itchat)详细介绍
Oct 23 Python
Python编程求质数实例代码
Jan 31 Python
PYQT5实现控制台显示功能的方法
Jun 25 Python
把vgg-face.mat权重迁移到pytorch模型示例
Dec 27 Python
python3.7+selenium模拟淘宝登录功能的实现
May 26 Python
python logging模块的使用
Sep 07 Python
Python实现Appium端口检测与释放的实现
Dec 31 Python
matplotlib绘制正余弦曲线图的实现
Feb 22 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安装攻略:常见问题解答(一)
2006/10/09 PHP
phpfans留言版用到的install.php
2007/01/04 PHP
简单谈谈php中ob_flush和flush的区别
2014/11/27 PHP
PHP中strncmp()函数比较两个字符串前2个字符是否相等的方法
2016/01/07 PHP
使用自定义setTimeout和setInterval使之可以传递参数和对象参数
2009/04/24 Javascript
JavaScript弹簧振子超简洁版 完全符合能量守恒,胡克定理
2009/10/25 Javascript
javascript中的prototype属性使用说明(函数功能扩展)
2010/08/16 Javascript
基于jquery的一个浮动框(扩展性比较好 )
2010/08/27 Javascript
js运动框架_包括图片的淡入淡出效果
2013/05/11 Javascript
JS图片根据鼠标滚动延时加载的实例代码
2013/07/13 Javascript
javascript使用prototype完成单继承
2014/12/24 Javascript
jQuery网页选项卡插件rTabs用法实例分析
2015/08/26 Javascript
JavaScript中对DOM节点的访问、创建、修改、删除
2015/11/16 Javascript
jQuery实现图片局部放大镜效果
2016/03/17 Javascript
懒加载实现的分页&&网站footer自适应
2016/12/21 Javascript
JS表格组件神器bootstrap table使用指南详解
2017/04/12 Javascript
AngularJs 延时器、计时器实例代码
2017/09/16 Javascript
vue实现Excel文件的上传与下载功能的两种方式
2019/06/28 Javascript
JS实现简单的表格增删
2020/01/16 Javascript
使用python调用浏览器并打开一个网址的例子
2014/06/05 Python
Python中字符串的修改及传参详解
2016/11/30 Python
Python中常用信号signal类型实例
2018/01/25 Python
Python 利用内置set函数对字符串和列表进行去重的方法
2018/06/29 Python
python实现图片彩色转化为素描
2019/01/15 Python
python实现nao机器人手臂动作控制
2019/04/29 Python
python实现横向拼接图片
2020/03/23 Python
pytorch 中的重要模块化接口nn.Module的使用
2020/04/02 Python
解决Python spyder显示不全df列和行的问题
2020/04/20 Python
Django基于Models定制Admin后台实现过程解析
2020/11/11 Python
10个很棒的 CSS3 开发工具 推荐
2011/05/16 HTML / CSS
英国优质家居用品网上品牌:URBANARA
2018/06/01 全球购物
Otiumberg官网:英国半精致珠宝品牌
2021/01/16 全球购物
2014领导班子“四风问题”对照检查材料思想汇报(执法局)
2014/09/21 职场文书
运动员加油词
2015/07/18 职场文书
高二化学教学反思
2016/02/22 职场文书
手把手教你从零开始react+antd搭建项目
2021/06/03 Javascript