欢迎您访问365答案网,请分享给你的朋友!
生活常识 学习资料

关于tensorflow的dataset执行map的问题

时间:2023-08-19

对一个包含dataset的dataset,比如这种window过后的玩意儿
如果对这种dataset执行map,在map里对子元素的dataset进行遍历,就会出现内存泄漏并且卡死

import tensorflow as tfdef f(x:tf.data.Dataset): a=[i for i in x] return aa = tf.data.Dataset.range(9)b = a.window(3)c = b.map(f)print(list(c.as_numpy_iterator()))

目前不清楚原因,可能关于map的底层原理不太懂

如果想要把里面的dataset展开,必须使用flat_map

Copyright © 2016-2020 www.365daan.com All Rights Reserved. 365答案网 版权所有 备案号:

部分内容来自互联网,版权归原作者所有,如有冒犯请联系我们,我们将在三个工作时内妥善处理。