在Tensorflows功能API中以None，a，b形式引入张量

问题描述

我在Functional API中使用了tensorflow 通过形式的张量

(None,a,b)

在各层之间。对于自定义层，我想使用总是产生产品的基本操作 形式

(a,b)

为了拥有一致的解决方案，我也需要将其引入(None,b)形式。 我如何在当前版本的tensorflow中做到这一点。当前的许多解决方案都行不通，因为 占位符之类的概念似乎不再可用。

如果要添加自定义元素，此错误会使tensorflow Functional API失效。

解决方法

尝试在两者之间添加一个重塑图层， 检查：https://www.tensorflow.org/api_docs/python/tf/keras/layers/Reshape