如何在 tensorflow 中連接兩個具有動態形狀的張量？

Question

我有兩個張量我想在 tensorflow 中聯系。

所以這些張量的形狀是： (Bs,dynamics,n_features)

batch_size和n_features保證是相同的，所以我認為這個操作是可能的。

但是，我認為 tensorflow 會進行預檢查嗎？ （對不起，我是 tensorflow 的新手），tensorflow 將檢查兩個形狀為(None,None,n_features)的張量是否可以在軸 1 上連接，這是不可能的，因為它不知道None+None的結果是什么

有人對此有任何想法嗎？

Answer 1

在您的情況下，連接應該不是問題：

import tensorflow as tf

x1 = tf.random.normal((5, 10, 20))
x2 = tf.random.normal((5, 33, 20))

@tf.function
def concatenate(x1, x2):
  return tf.concat([x1, x2], axis=1)

concatenate(x1, x2).shape