2021年11月1日
在tensorflow.keras.layers.Layer自定义层中使用for循环
摘要： 踩坑事项： 一定要将for循环的次数在构建层时就传入__init__()，而不是和inputs一起使用call传入！特别是每轮循环都需要tf.concat()新张量的时候。否则tf构建动态图无法计算循环后数据的维度报错！ 阅读全文
posted @ 2021-11-01 17:21 风和雨滴 阅读(247) 评论(0) 推荐(0) 编辑