在tensorflows函数API中引入一个(None,a,b)形式的张量



我正在将tensorflow与Functional API一起使用它通过形式的张量

(None, a,b)

在层之间。对于自定义层,我希望使用始终产生产品的基本操作形式

(a,b)

为了有一个一致的解决方案,我也需要将其形式化为(None,a,b)。如何在当前版本的tensorflow中做到这一点。目前的许多解决方案都不起作用,因为像占位符这样的概念似乎已经不可用了。

如果想要添加自定义元素,则此错误会使tensorflow Functional API变得无用。

尝试在两者之间添加一个整形层,检查:https://www.tensorflow.org/api_docs/python/tf/keras/layers/Reshape

最新更新