我需要执行一个足够简单的TensorFlow操作,但我想不出该怎么做。
当前,我的数据具有以下格式的张量:
[[a0], [a1], [a2], ...]
我想将该数据转换为以下格式:
[[1-a0, a0], [1-a1, a1], [1-a2, a2], ...]
可能有一个非常简单的解决方案,我缺少词汇来发现-如果可以直接使用它,将不胜感激。
答案 0 :(得分:2)
我认为您可以使用tf.concat
:
tf.concat([1 - x, x], axis=1)
# <tf.Tensor 'concat:0' shape=(3, 2) dtype=int32>