Tensorflow：串聯包含無形狀的2個張量

Question

我的問題如下：我有一個形狀爲[None, None, None, 3]（[batch_size, height, width, num_channels]）和張量B的張量A。在運行時，它是保證A和B將具有相同的形狀。我想沿着num_channels軸連接這兩個張量。

PS。請注意，我簡化了原來的問題 - 因此首先形狀[None, None, None, 6]的張量不是一個選項。

Answer 1

在問一個問題之前，我應該嘗試運行代碼:)使用tf.concat(values=[A, B], concat_dim=3)似乎正在工作。