Тензорный поток: конкатенация 2 тензора форм, содержащих Нет

Question

Моя проблема заключается в следующем: у меня есть тензор A формы [None, None, None, 3] ([batch_size, height, width, num_channels]) и тензор B. Во время работы гарантирует, что A и B будут иметь одинаковую форму. Я хотел бы объединить эти два тензора вдоль оси num_channels.

PS. Обратите внимание, что я упростил мою первоначальную проблему - поэтому наличие тензора формы [None, None, None, 6] в первую очередь не является вариантом.

Answer 1

Прежде чем задать вопрос, я должен, вероятно, попытаться запустить код :) Использование tf.concat(values=[A, B], concat_dim=3) похоже работает.