如何在张量流中间隔连接两个张量？

Question

我想在tensorflow2中将两个张量连接成棋盘格状，如下所示：

示例1：

a = [[1,1],[1,1]]
b = [[0,0],[0,0]]

concated_a_and_b = [[1,0,1,0],[0,1,0,1]]

示例2：

a = [[1,1,1],[1,1,1],[1,1,1]]
b = [[0,0,0],[0,0,0],[0,0,0]]

concated_a_and_b = [[1,0,1,0,1,0],[0,1,0,1,0,1],[1,0,1,0,1,0]]

tensorflow2中是否有一种像这样的方式将它们串联起来的方法？

一些背景知识： 我首先将带有棋盘蒙版的张量c分成两个半a和b。经过一番改造之后，我不得不将它们重新组合成原始的形状和顺序。

我所说的棋盘格意思是：

Answer 1

步骤1：生成具有交替值的矩阵

您可以先将它们串联成[1，0]对，然后再施加最终的整形。

第2步：撤消一些行

我将矩阵分为两部分，反转第二部分，然后通过从第一部分和第二部分中进行选择来重建完整的矩阵

代码示例：

import math
import numpy as np
import tensorflow as tf

a = tf.ones(shape=(3, 4))
b = tf.zeros(shape=(3, 4))

x = tf.expand_dims(a, axis=-1)
y = tf.expand_dims(b, axis=-1)

paired_ones_zeros = tf.concat([x, y], axis=-1)

alternated_values = tf.reshape(paired_ones_zeros, [-1, a.shape[1] + b.shape[1]])

num_samples = alternated_values.shape[0]
middle = math.ceil(num_samples / 2)
is_num_samples_odd = middle * 2 != num_samples

# Gather first part of the matrix, don't do anything to it
first_elements = tf.gather_nd(alternated_values, [[index] for index in range(middle)])
# Gather second part of the matrix and reverse its elements
second_elements = tf.reverse(tf.gather_nd(alternated_values, [[index] for index in range(middle, num_samples)]), axis=[1])

# Pick alternatively between first and second part of the matrix
indices = np.concatenate([[[index], [index + middle]] for index in range(middle)], axis=0)
if is_num_samples_odd:
    indices = indices[:-1]

output = tf.gather_nd(
    tf.concat([first_elements, second_elements], axis=0),
    indices
)
print(output)

Answer 2

我知道这不是一个体面的方法，因为它会影响时间和空间的复杂性。但这解决了以上问题

   def concat(tf1, tf2):
      result = []
      for (index, (tf_item1, tf_item2)) in enumerate(zip(tf1, tf2)):
        item = []
        for (subitem1, subitem2) in zip(tf_item1, tf_item2):
            if index % 2 == 0:
                item.append(subitem1)
                item.append(subitem2)
            else:
                item.append(subitem2)
                item.append(subitem1)
        concated_a_and_b.append(item)
      return concated_a_and_b