Reverse jede zweite Zeile in TensorFlow

Question

Gegeben ein Tensor input der undefinierten Form H x W, würde ich gerne jede zweite Zeile umkehren.

In numpy, würde ich einfach tun

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

aber das ist offenbar nicht möglich, in TensorFlow.

Beachten Sie, dass die Eingabeform nur teilweise bekannt ist, d.h. input.shape == (None, None).

Irgendwelche Ideen?

Answer 1

Sie können das gleiche mit Platzhalter

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})