Tensorflow：连接 2 个包含 None 的形状张量

Question

我的问题如下：我有一个形状为 [None, None, None, 3] ([batch_size, height, width, num_channels]) 的张量 A 和一个张量 B。在运行时，保证A 和B 将具有相同的形状。我想将这两个张量沿num_channels 轴连接起来。

附言。请注意，我简化了我最初的问题 - 所以首先不能选择形状为 [None, None, None, 6] 的张量。

Answer 1

在提出问题之前，我可能应该尝试运行代码 :) 使用 tf.concat(values=[A, B], concat_dim=3) 似乎有效。