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 字符串split的用法分享
Mar 23 Python
Python CSV模块使用实例
Apr 09 Python
Python实现XML文件解析的示例代码
Feb 05 Python
Python测试人员需要掌握的知识
Feb 08 Python
Python3+PyInstall+Sciter解决报错缺少dll、html等文件问题
Jul 15 Python
FFrpc python客户端lib使用解析
Aug 24 Python
Pycharm+django2.2+python3.6+MySQL实现简单的考试报名系统
Sep 05 Python
Keras预训练的ImageNet模型实现分类操作
Jul 07 Python
如何快速理解python的垃圾回收机制
Sep 01 Python
如何使用python写截屏小工具
Sep 29 Python
Python操作CSV格式文件的方法大全
Jul 15 Python
基于python定位棋子位置及识别棋子颜色
Jul 26 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
Protoss兵种介绍
2020/03/14 星际争霸
dedecms模板标签代码官方参考
2007/03/17 PHP
php判断并删除空目录及空子目录的方法
2015/02/11 PHP
PHP文件系统管理(实例讲解)
2017/09/19 PHP
PHP 与 js的通信(via ajax,json)
2010/11/16 Javascript
MooBox 基于Mootools的对话框插件
2012/01/20 Javascript
js页面跳转的常用方法整理
2013/10/18 Javascript
js 自动播放的实例代码
2013/11/19 Javascript
jquery和ajax的关系详细介绍
2013/11/29 Javascript
JavaScript检查某个function是否是原生代码的方法
2014/08/20 Javascript
javascript使用call调用微信API
2014/12/15 Javascript
JavaScript实现简单动态进度条效果
2018/04/06 Javascript
Vue获取页面元素的相对位置的方法示例
2020/02/05 Javascript
详解vue中v-model和v-bind绑定数据的异同
2020/08/10 Javascript
[02:26]DOTA2英雄米拉娜基础教程
2013/11/25 DOTA
Python抓取手机号归属地信息示例代码
2016/11/28 Python
Python 模块EasyGui详细介绍
2017/02/19 Python
python模拟登录并且保持cookie的方法详解
2017/04/04 Python
Linux 修改Python命令的方法示例
2018/12/03 Python
PyQt编程之如何在屏幕中央显示窗体的实例
2019/06/18 Python
python多线程高级锁condition简单用法示例
2019/11/07 Python
pytorch实现建立自己的数据集(以mnist为例)
2020/01/18 Python
解决Python logging模块无法正常输出日志的问题
2020/02/21 Python
Python操作Word批量生成合同的实现示例
2020/08/28 Python
微信浏览器取消缓存的方法
2015/03/28 HTML / CSS
马来西亚排名第一的宠物用品店:Pets Wonderland
2020/04/16 全球购物
威盛公司软件C++工程师笔试题面试题
2012/07/16 面试题
开放系统互连参考模型
2016/06/29 面试题
军训心得体会
2013/12/31 职场文书
热门专业求职信
2014/05/24 职场文书
二年级语文下册复习计划
2015/01/19 职场文书
活着观后感
2015/06/03 职场文书
汤姆索亚历险记读书笔记
2015/06/29 职场文书
2016中秋节广告语
2016/01/28 职场文书
HTML中的表单Form实现居中效果
2021/05/25 HTML / CSS
Python中的嵌套循环详情
2022/03/23 Python