Другой многомерный конкат в TensorFlow

Question

Мне нужно объединить эти тензоры в измерениях 2 и 3:

[<tf.Tensor 'one' shape=(3, 3, 15, 30) dtype=float32>, 
<tf.Tensor 'two' shape=(3, 3, 14, 29) dtype=float32>,
<tf.Tensor 'three' shape=(3, 3, 1, 1) dtype=float32>]

Я знаю, что в пространстве это не имеет смысла, но я попытался преобразовать в:

[<tf.Tensor 'one' shape=(3, 3, xx) dtype=float32>, 
<tf.Tensor 'two' shape=(3, 3, yy) dtype=float32>,
<tf.Tensor 'three' shape=(3, 3, zz) dtype=float32>]

А потом конкат:

[<tf.Tensor 'one' shape=(3, 3, xx+yy+zz) dtype=float32>]

Имеет ли это смысл? У кого-нибудь есть другая идея?

Answer 1

Вы можете использовать tf.concat с параметром оси, установленным в 2.

x = [tensor1, tensor2, tensor3]
tf.concat(x, axis=2)