可以使用tf.concat获得组合元组张量?

时间:2018-05-25 01:53:11

标签: python tensorflow tensor

我显然是张力流的新手,并尝试使用两个现有的张量来制作新的张量。

让我们说有两个张量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)]

因此,第一张量中的第一项与第二张量中的第一项相结合,并将组合项转换为元组。

1 个答案:

答案 0 :(得分:0)

我可以建议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相当清洁。