tf.unstack用法_jiuweideqixu的博客

import tensorflow as tf
a = tf.constant([[1,2,3],[4,5,6]])

ar = tf.reshape(a,[2,1,3])

with tf.Session() as sess:
    print(sess.run(a))
    print('a<-------------------->ar')
    print(sess.run(ar))
print('======================')
aru = tf.unstack(ar,axis=1)
au = tf.unstack(a,axis=0)

with tf.Session() as sess:
    print(sess.run(aru))
    print('aru<----------------->au')
    print(sess.run(au))import tensorflow as tf
a = tf.constant([[1,2,3