給定張量input
未定義的形狀H x W
,我想反轉其他所有行。在TensorFlow中每隔一行反向
在numpy的,我只想做
input[1::2, :] = input[1::2, ::-1]
但這顯然是不可能的TensorFlow。
請注意,輸入形狀僅爲部分已知的,即input.shape == (None, None)
。
任何想法?
給定張量input
未定義的形狀H x W
,我想反轉其他所有行。在TensorFlow中每隔一行反向
在numpy的,我只想做
input[1::2, :] = input[1::2, ::-1]
但這顯然是不可能的TensorFlow。
請注意,輸入形狀僅爲部分已知的,即input.shape == (None, None)
。
任何想法?
可以實現使用佔位符
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})
同樣是已知的偶數行的數量,或者是亂? – user1735003
@ user1735003它是任意的:)它使得它更難 – fabian789