在tensorflow.keras.layers.Layer自定义层中使用for循环

踩坑事项:

  • 一定要将for循环的次数在构建层时就传入__init__(),而不是和inputs一起使用call传入!特别是每轮循环都需要tf.concat()新张量的时候。否则tf构建动态图无法计算循环后数据的维度报错!
posted @ 2021-11-01 17:21  风和雨滴  阅读(247)  评论(0编辑  收藏  举报