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中操作字典之clear()方法的使用
May 21 Python
利用python解决mysql视图导入导出依赖的问题
Dec 17 Python
对python 操作solr索引数据的实例详解
Dec 07 Python
python保存字典和读取字典的实例代码
Jul 07 Python
python中bs4.BeautifulSoup的基本用法
Jul 27 Python
Python绘制热力图示例
Sep 27 Python
Python中join()函数多种操作代码实例
Jan 13 Python
python3+opencv生成不规则黑白mask实例
Feb 19 Python
python字符串常用方法及文件简单读写的操作方法
Mar 04 Python
Python数据可视化实现漏斗图过程图解
Jul 20 Python
pytorch 预训练模型读取修改相关参数的填坑问题
Jun 05 Python
Python中的matplotlib绘制百分比堆叠柱状图,并为每一个类别设置不同的填充图案
Apr 20 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
超人钢铁侠联手合作?美漫作家呼吁DC漫威合作联动以抵抗疫情
2020/04/09 欧美动漫
PHP 程序授权验证开发思路
2009/07/09 PHP
PHP 分页类(模仿google)-面试题目解答
2009/09/13 PHP
discuz authcode 经典php加密解密函数解析
2020/07/12 PHP
PHP统一页面编码避免乱码问题
2015/04/09 PHP
编写PHP脚本过滤用户上传的图片
2015/07/03 PHP
利用PHP生成CSV文件简单示例
2016/12/21 PHP
PHP获取路径和目录的方法总结【必看篇】
2017/03/04 PHP
javascript的onchange事件与jQuery的change()方法比较
2009/09/28 Javascript
Javascript常考语句107条收集
2010/03/09 Javascript
JavaScript中的console.dir()函数介绍
2014/12/29 Javascript
window.setInterval()方法的定义和用法及offsetLeft与style.left的区别
2015/11/11 Javascript
跟我学习javascript的arguments对象
2015/11/16 Javascript
js实现内容显示并使用json传输数据
2016/03/16 Javascript
浅谈JavaScript for循环 闭包
2016/06/22 Javascript
利用jQuery.Validate异步验证用户名是否存在(推荐)
2016/12/09 Javascript
基于Vue的SPA动态修改页面title的方法(推荐)
2018/01/02 Javascript
Vuex 进阶之模块化组织详解
2018/01/12 Javascript
vue.js计算属性computed用法实例分析
2018/07/06 Javascript
20多个小事例带你重温ES10新特性(小结)
2019/09/29 Javascript
详解将Pandas中的DataFrame类型转换成Numpy中array类型的三种方法
2019/07/06 Python
django重新生成数据库中的某张表方法
2019/08/28 Python
pytorch中获取模型input/output shape实例
2019/12/30 Python
接口自动化多层嵌套json数据处理代码实例
2020/11/20 Python
python 利用百度API识别图片文字(多线程版)
2020/12/14 Python
英国HYPE双肩包官网:英国本土时尚潮牌
2018/09/26 全球购物
ECCO俄罗斯官网:北欧丹麦鞋履及皮具品牌
2020/06/26 全球购物
Sql面试题
2013/03/20 面试题
青年创业培训欢迎词
2014/01/08 职场文书
期末复习计划
2015/01/19 职场文书
城南旧事读书笔记
2015/06/29 职场文书
2016关于读书活动的心得体会
2016/01/14 职场文书
php微信小程序解包过程实例详解
2021/03/31 PHP
Python 数据可视化之Seaborn详解
2021/11/02 Python
Python数据结构之队列详解
2022/03/21 Python
APP界面设计技巧和注意事项
2022/04/29 杂记