可以使用 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 比较干净。