如何在 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