我将tensorflow与Functional API一起使用,它传递表单的张量
(None, a,b)
层与层之间。对于自定义层,我希望使用基本操作,这些操作总是生成以下形式的产品
(a,b)
为了获得一致的解决方案,我还需要将其引入到表单(None,a,b)
中。如何在当前版本的tensorflow中做到这一点。许多当前的解决方案都不起作用,因为占位符等概念似乎不再可用。
如果要添加自定义元素,此错误将使tensorflow函数式API变得无用。
发布于 2020-10-06 02:43:20
尝试在中间添加一个重塑图层,检查:https://www.tensorflow.org/api_docs/python/tf/keras/layers/Reshape
https://stackoverflow.com/questions/64218513
复制