可以使用 tf.concat 获得组合的元组张量吗？

Question

我显然是 tensorflow 的新手，并试图使用两个现有的张量制作新的张量。

假设有两个张量 t1 和 t2。

t1 = [1, 2, 3, 4, 5, 6]
t2 = [7, 8, 9, 10, 11, 12]

无论如何我可以使用 tf.concat 获得如下形状的新张量 t3？

t3 = [(1,7),(2,8),(3,9),(4,10),(5,11),(6,12)]

所以第一个张量中的第一项与第二个张量中的第一项合并，并将合并的项转换为一个元组。

Answer 1

我可以推荐tf.stack吗？

tf.stack((t1, t2), axis=1)

如果您打算使用tf.concat，您可以提前扩展到一个额外的维度吗？

tf.concat((tf.expand_dims(t1, 1), tf.expand_dims(t2, 1)), axis=1)

但是tf.stack 干净多了。