2017-08-02 45 views
0

給定張量input未定義的形狀H x W,我想反轉其他所有行。在TensorFlow中每隔一行反向

在numpy的,我只想做

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

但這顯然是不可能的TensorFlow。

請注意,輸入形狀僅爲部分已知的,即input.shape == (None, None)

任何想法?

+0

同樣是已知的偶數行的數量,或者是亂? – user1735003

+0

@ user1735003它是任意的:)它使得它更難 – fabian789

回答

0

可以實現使用佔位符

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

但是這反轉了每一行,對吧? – fabian789

+0

是的每一行。我認爲那是你的問題? –

+0

不,這是隔行:) – fabian789