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深入学习之特殊方法与多范式
Aug 31 Python
Pycharm学习教程(1) 定制外观
May 02 Python
Python实现的生产者、消费者问题完整实例
May 30 Python
redis之django-redis的简单缓存使用
Jun 07 Python
Python求离散序列导数的示例
Jul 10 Python
下载官网python并安装的步骤详解
Oct 12 Python
浅谈Python程序的错误:变量未定义
Jun 02 Python
使用K.function()调试keras操作
Jun 17 Python
通过代码实例了解Python异常本质
Sep 16 Python
python 用opencv实现霍夫线变换
Nov 27 Python
python办公自动化之excel的操作
May 23 Python
python opencv检测直线 cv2.HoughLinesP的实现
Jun 18 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
function, new function, new Function之间的区别
2007/03/08 Javascript
Jquery Ajax 学习实例2 向页面发出请求 返回JSon格式数据
2010/03/15 Javascript
js 兼容多浏览器的回车和鼠标焦点事件代码(IE6/7/8,firefox,chrome)
2010/04/14 Javascript
JSChart轻量级图形报表工具(内置函数中文参考)
2010/10/11 Javascript
js中关于String对象的replace使用详解
2011/05/24 Javascript
JQuery UI的拖拽功能实现方法小结
2012/03/14 Javascript
用js调用迅雷下载代码的二种方法
2013/04/15 Javascript
ajax如何实现页面局部跳转与结果返回
2015/08/24 Javascript
JavaScript中的boolean布尔值使用学习及相关技巧讲解
2016/05/26 Javascript
jQuery+ajax实现实用的点赞插件代码
2016/07/06 Javascript
jQuery Raty 一款不错的星级评分插件
2016/08/24 Javascript
Javascript日期格式化format函数的使用方法
2016/08/30 Javascript
深入浅出es6模板字符串
2017/08/26 Javascript
JavaScript实现淘宝京东6位数字支付密码效果
2018/08/18 Javascript
React 组件中的 bind(this)示例代码
2018/09/16 Javascript
详解小程序如何避免多次点击,重复触发事件
2019/04/08 Javascript
微信小程序页面调用自定义组件内的事件详解
2019/09/12 Javascript
vue实现在线学生录入系统
2020/05/30 Javascript
vue集成一个支持图片缩放拖拽的富文本编辑器
2021/01/29 Vue.js
关于better-scroll插件的无法滑动bug(2021通过插件解决)
2021/03/01 Javascript
Python中字典(dict)和列表(list)的排序方法实例
2014/06/16 Python
以一段代码为实例快速入门Python2.7
2015/03/31 Python
python判断字符串编码的简单实现方法(使用chardet)
2016/07/01 Python
python解析基于xml格式的日志文件
2017/02/25 Python
详解python 拆包可迭代数据如tuple, list
2017/12/29 Python
Django中使用Celery的方法示例
2018/11/29 Python
详解从Django Allauth中进行登录改造小结
2019/12/18 Python
使用python+poco+夜神模拟器进行自动化测试实例
2020/04/23 Python
Pandas实现一列数据分隔为两列
2020/05/18 Python
Python OpenCV实现测量图片物体宽度
2020/05/27 Python
浅析HTML5页面元素及属性
2021/01/20 HTML / CSS
函授教育个人学习的自我评价
2013/12/31 职场文书
大学运动会通讯稿
2014/01/28 职场文书
服务之星事迹材料
2014/05/03 职场文书
物业公司管理制度
2015/08/05 职场文书
python基于tkinter制作无损音乐下载工具
2021/03/29 Python