tensorflow 和keras 转化的时候

  • Post author:
  • Post category:其他


keras  :

a = Input( shape=(1,))

a 的shape为(?,1)

tensorflow :

a = tf.placeholder(
    shape=(1,), name=prefix + fc.name, dtype=fc.dtype)

生成的a 为 (1,)

应该转化为

a = tf.placeholder(
    shape=(None,1), name=prefix + fc.name, dtype=fc.dtype)



版权声明:本文为qq_40210472原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。