Tensorflow: Verketten von 2 Tensoren von Formen, die keine enthalten

Question

Mein Problem ist folgendes: Ich habe einen Tensor A der Form [None, None, None, 3] ([batch_size, height, width, num_channels]) und einen Tensor B. Zur Laufzeit ist es garantiert, dass A und B die gleiche Form haben werden. Ich möchte diese beiden Tensoren entlang der Achse num_channels verketten.

PS. Beachten Sie, dass ich mein ursprüngliches Problem vereinfacht habe - es ist also keine Option, einen Tensor der Form [None, None, None, 6] zu haben.

Answer 1

Bevor ich eine Frage stelle, sollte ich wahrscheinlich versuchen, den Code auszuführen :) Die Verwendung von tf.concat(values=[A, B], concat_dim=3) scheint zu funktionieren.