Concat两个2D张量

时间:2019-06-26 01:27:47

标签: python python-3.x tensorflow tensor

我需要执行一个足够简单的TensorFlow操作,但我想不出该怎么做。

当前,我的数据具有以下格式的张量:

[[a0], [a1], [a2], ...]

我想将该数据转换为以下格式:

[[1-a0, a0], [1-a1, a1], [1-a2, a2], ...]

可能有一个非常简单的解决方案,我缺少词汇来发现-如果可以直接使用它,将不胜感激。

1 个答案:

答案 0 :(得分:2)

我认为您可以使用tf.concat

tf.concat([1 - x, x], axis=1)
# <tf.Tensor 'concat:0' shape=(3, 2) dtype=int32>