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



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

假设有两个张量 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)]

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

我可以建议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要干净得多。

最新更新