Tensorflow: Using tf.slice to split the input

Question

I\'m trying to split my input layer into different sized parts. I\'m trying to use tf.slice to do that but it\'s not working.

Some sample code:

import te         


        

Answer 1

You can also try out this one

x = tf.slice(ph, [0,0], [3, 2])

As your starting point is (0,0) second argument is [0,0]. You want to slice three raw and two column so your third argument is [3,2].

This will give you desired output.