在TensorFlow中反转每隔一行

Question

给定未定义形状input的张量H x W，我想反转所有其他行。

在numpy中，我只会做

input[1::2, :] = input[1::2, ::-1]

但这在TensorFlow中显然是不可能的。

请注意，输入形状仅为部分已知，即input.shape == (None, None)。

有什么想法吗？

Answer 1

使用占位符

可以实现相同的效果

input = tf.placeholder(shape=(None, None), dtype=tf.int32)
# define axis to reverse
axis_to_reverse=1
input_reversed = tf.reverse(input, [axis_to_reverse])

sess = tf.Session()
_input_reversed = sess.run(input_reversed, {input: your array})