【Tensorflow】tf.stack() tf.unstack()

一。

1. import tensorflow as tf  
2.   
3. a = tf.constant([1,2,3])  
4. b = tf.constant([4,5,6])  
5.   
6. c = tf.stack([a,b],axis=0)  
7. d = tf.stack([a,b],axis=1)  
8.   
9. e = tf.unstack(d,axis=0)  
10. f = tf.unstack(d,axis=1)  
11.   
12.   
13. with tf.Session() as sess:  
14.     print('原始向量数据：')  
15.     print(sess.run(a))  
16.     print(sess.run(b))       
17.     print('矩阵拼接的函数示例,得到一个矩阵：')  #返回值是多维矩阵  
18.     print('以"0维"的方式进行拼接')  
19.     print(sess.run(c))  
20.     print('以"1维"的方式进行拼接')  
21.     print(sess.run(d))  
22.       
23.     print('矩阵分解的函数示例，得到一个list：') #返回值是一个list     
24.     print('以"0维"的方式进行分解')  
25.     print(sess.run(e))  
26.     print('以"1维"的方式进行分解')  
27.     print(sess.run(f)) 
    

二。