小编wwi*_*ook的帖子

我正在尝试在 Tensorflow 2.0RC 中实现自定义 Keras Layer，并且需要将[None, Q]成形张量连接到成形张量上[None, H, W, D]以生成[None, H, W, D + Q]成形张量。假设两个输入张量具有相同的批量大小，即使事先未知。此外，H、W、D 和 Q 在写入时都是未知的，但build在首次调用层时在层的方法中进行评估。我遇到的问题是当将[None, Q]成形张量广播到[None, H, W, Q]成形张量以便连接时。

以下是尝试使用功能 API 创建 Keras 的示例，该 API 执行从形状到形状Model的可变批量广播：[None, 3][None, 5, 5, 3]

import tensorflow as tf
import tensorflow.keras.layers as kl
import numpy as np

x = tf.keras.Input([3])  # Shape [None, 3]
y = kl.Reshape([1, 1, 3])(x)  # Need to add empty dims …