tensorflow 取 sub-tensor 实例

import tensorflow as tf

value = [[0,1],[2,3],[4,5],[6,7]]
init = tf.constant_initializer(value)
v = tf.get_variable('value', shape=[4,2], initializer=init,dtype=tf.int32)


v_ = v[:3,:]

init = tf.initialize_all_variables()
sess = tf.Session()
sess.run(init)
print(sess.run(v_))