矩阵拼接的函数tf.stack()与矩阵分解的函数tf.unstack()示例

import tensorflow as tf

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

c = tf.stack([a,b],axis=0)
d = tf.stack([a,b],axis=1)

e = tf.unstack(d,axis=0)
f = tf.unstack(d,axis=1)


with tf.Session() as sess:
    print('原始向量数据：')
    print(sess.run(a))
    print(sess.run(b))     
    print('矩阵拼接的函数示例,得到一个矩阵：')  #返回值是多维矩阵
    print('以"0维"的方式进行拼接')
    print(sess.run(c))
    print('以"1维"的方式进行拼接')
    print(sess.run(d))
    
    print('矩阵分解的函数示例，得到一个list：') #返回值是一个list   
    print('以"0维"的方式进行分解')
    print(sess.run(e))
    print('以"1维"的方式进行分解')
    print(sess.run(f))