以交替方式連接兩個張量（Tensorflow）

Question

我想以交替方式連接兩個張量shape=(None, 16)（因此結果張量必須是shape=(None, 32)，其中第一張量的第一個數組與第一個張量交替混合我不能因爲未知shape[0]張量循環，不支持第二張之一等等。

我該怎麼辦呢？

zip函數張量（張量的對象不是可迭代） 我使用Python3的Tensorflow。

Answer 1

假設兩個張量在外部（None）維中具有相同的形狀，並且您想要在兩個張量的行之間交替，可以通過添加一個具有tf.expand_dims()的維，與tf.concat()連接，然後使用tf.reshape()重塑形：

# Use these tensors as example inputs, but the shape need not be statically known. 
x = tf.ones([37, 16]) 
y = tf.zeros([37, 16]) 

x_expanded = tf.expand_dims(x, 2)     # shape: (37, 16, 1) 
y_expanded = tf.expand_dims(y, 2)     # shape: (37, 16, 1) 

concatted = tf.concat([x_expanded, y_expanded], 2) # shape: (37, 16, 2) 

result = tf.reshape(concatted, [-1, 32])   # shape: (37, 32)