0
点赞
收藏
分享

微信扫一扫

对Keras中map函数和lambda的理解

四月天2021 2022-03-13 阅读 38

文章目录

遇到问题

今天在刷 deep learning with python second edition 的时候,在NLP章节遇到了自己没理解的知识:keras中map函数的用法

text_only_train_ds = train_ds.map(lambda x, y: x)

在这里插入图片描述

解决

这里map的作用是:让CPU可以异步计算。在GPU处理上一个batch的数据时,CPU已经在处理下一个batch的预处理了。这样GPU处理完这个batch之后,可以无需等待即可开始下一个batch的计算。

那么后面lambda表达式labmda x, y: x的作用是什么呢?lambda表达式会返回冒号后面的表达式,这里是x,也就是说返回x,丢弃了y(这里是标签)。

来自StackOverflow的问题

答案

参考

Map function in Keras

举报

相关推荐

0 条评论