keras Lambda自定义层实现数据的切片方式,Lambda传参数


Posted in Python onJune 11, 2020

1、代码如下:

import numpy as np
from keras.models import Sequential
from keras.layers import Dense, Activation,Reshape
from keras.layers import merge
from keras.utils.visualize_util import plot
from keras.layers import Input, Lambda
from keras.models import Model
 
def slice(x,index):
 return x[:,:,index]
 
a = Input(shape=(4,2))
x1 = Lambda(slice,output_shape=(4,1),arguments={'index':0})(a)
x2 = Lambda(slice,output_shape=(4,1),arguments={'index':1})(a)
x1 = Reshape((4,1,1))(x1)
x2 = Reshape((4,1,1))(x2)
output = merge([x1,x2],mode='concat')
model = Model(a, output)
x_test = np.array([[[1,2],[2,3],[3,4],[4,5]]])
print model.predict(x_test)
plot(model, to_file='lambda.png',show_shapes=True)

2、注意Lambda 是可以进行参数传递的,传递的方式如下代码所述:

def slice(x,index):
return x[:,:,index]

如上,index是参数,通过字典将参数传递进去.

x1 = Lambda(slice,output_shape=(4,1),arguments={'index':0})(a)
x2 = Lambda(slice,output_shape=(4,1),arguments={'index':1})(a)

3、上述代码实现的是,将矩阵的每一列提取出来,然后单独进行操作,最后在拼在一起。可视化的图如下所示。

keras Lambda自定义层实现数据的切片方式,Lambda传参数

补充知识:tf.keras.layers.Lambda()——匿名函数层解析

1. 参数列表

keras Lambda自定义层实现数据的切片方式,Lambda传参数

2. 作用

keras Lambda自定义层实现数据的切片方式,Lambda传参数

可以把任意的一个表达式作为一个“Layer”对象

Lambda层之所以存在是因为它可以在构建Squential时使用任意的函数或者说tensorflow 函数。

在我们需要完成一些简单的操作(例如VAE中的重采样)的情况下,Lambda层再适合不过了。

3. 举个栗子(VAE)

可以看到通过在encoder和decoder中间加入一个Lambda层使得encoder和decoder连接起来,很方便

def sampling(agrs):
  mean,logvar = agrs[0],agrs[1]
  eps = tf.random.normal(tf.shape(mean))
  return mean + eps*tf.exp(logvar * 0.5)

# 编码阶段
  
x = layers.Input(shape=(784,)) # 输入层
  
h1 = layers.Dense(200,activation='softplus')(x)
h2 = layers.Dense(200,activation='softplus')(h1)
# 均值和方差层不需要激活函数
mean = layers.Dense(latent_dimension)(h2)
log_var = layers.Dense(latent_dimension)(h2)
  
# 将采样过程看成一个Lambda层,这里利用自定义的sampling函数
z = layers.Lambda(sampling,output_shape=(latent_dimension,))([mean,log_var])
  
# 解码阶段
h3 = layers.Dense(200,activation='softplus')
h4 = layers.Dense(200,activation='softplus')
h5 = layers.Dense(200,activation='softplus')
# No activation
end = layers.Dense(784)
z1 = h3(z)
z2 = h4(z1)
z3 = h5(z2)
out = end(z3)
  
# 建立模型
model = tf.keras.Model(x,out)

4. Lambda层的缺点

Lambda层虽然很好用,但是它不能去更新我们模型的配置信息,就是不能重写'model.get_config()'方法

所以tensorflow提议,尽量使用自定义层(即tf.keras.layers的子类)

关于自定义层,我的博客有一期会专门讲

总结

当网络需要完成一些简单的操作时,可以考虑使用Lambda层。

以上这篇keras Lambda自定义层实现数据的切片方式,Lambda传参数就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python警察与小偷的实现之一客户端与服务端通信实例
Oct 09 Python
python实现ping的方法
Jul 06 Python
python3实现字符串的全排列的方法(无重复字符)
Jul 07 Python
Python 移动光标位置的方法
Jan 20 Python
详解PANDAS 数据合并与重塑(join/merge篇)
Jul 09 Python
Django中自定义查询对象的具体使用
Oct 13 Python
Python解析json代码实例解析
Nov 25 Python
django框架cookie和session用法实例详解
Dec 10 Python
python cv2读取rtsp实时码流按时生成连续视频文件方式
Dec 25 Python
keras 多任务多loss实例
Jun 22 Python
提高python代码运行效率的一些建议
Sep 29 Python
方法汇总:Python 安装第三方库常用
Apr 26 Python
python怎么提高计算速度
Jun 11 #Python
Python 实现自动登录+点击+滑动验证功能
Jun 10 #Python
Python函数参数定义及传递方式解析
Jun 10 #Python
什么是python类属性
Jun 10 #Python
基于SQLAlchemy实现操作MySQL并执行原生sql语句
Jun 10 #Python
浅谈Python中的模块
Jun 10 #Python
keras多显卡训练方式
Jun 10 #Python
You might like
一个程序下载的管理程序(三)
2006/10/09 PHP
Yii框架form表单用法实例
2014/12/04 PHP
php连接oracle数据库及查询数据的方法
2014/12/29 PHP
thinkPHP学习笔记之安装配置篇
2015/03/05 PHP
Yii2框架实现注册和登录教程
2016/09/30 PHP
php实现socket推送技术的示例
2017/12/20 PHP
用js实现手把手教你月入万刀(转贴)
2007/11/07 Javascript
游览器中javascript的执行过程(图文)
2012/05/20 Javascript
js AppendChild与insertBefore用法详细对比
2013/12/16 Javascript
jquery动态改变div宽度和高度
2015/02/09 Javascript
jQuery实现冻结表头的方法
2015/03/09 Javascript
JavaScript检测字符串中是否含有html标签实现方法
2015/07/01 Javascript
JavaScript各类型的关系图解
2015/10/16 Javascript
javascript完美实现给定日期返回上月日期的方法
2017/06/15 Javascript
JavaScript中的一些实用小技巧总结
2019/04/07 Javascript
Nodejs libuv运行原理详解
2019/08/21 NodeJs
js实现坦克移动小游戏
2019/10/28 Javascript
基于Electron实现桌面应用开发代码实例
2020/07/07 Javascript
vue data变量相互赋值后被实时同步的解决步骤
2020/08/05 Javascript
原生js实现俄罗斯方块
2020/10/20 Javascript
Python数组遍历的简单实现方法小结
2016/04/27 Python
举例讲解Python编程中对线程锁的使用
2016/07/12 Python
Python实现视频下载功能
2017/03/14 Python
python实现栅栏加解密 支持密钥加密
2019/03/20 Python
完美解决pycharm导入自己写的py文件爆红问题
2020/02/12 Python
Python生成并下载文件后端代码实例
2020/08/31 Python
python+selenium自动化实战携带cookies模拟登陆微博
2021/01/19 Python
html5跳转小程序wx-open-launch-weapp踩坑
2020/12/02 HTML / CSS
自我鉴定思想方面
2013/10/07 职场文书
舞蹈教育学专业推荐信
2013/11/27 职场文书
公司部门司机岗位职责
2014/01/03 职场文书
学校领导班子四风对照检查材料
2014/09/27 职场文书
校长师德师风自我剖析材料
2014/09/29 职场文书
学生早退检讨书(范文)
2019/08/19 职场文书
彻底理解golang中什么是nil
2021/04/29 Golang
javascript对象3个属性特征
2021/11/17 Javascript