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

来源:互联网 发布:三级分销佣金软件 编辑:程序博客网 时间:2024/06/07 00:49
import tensorflow as tfa = 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))

原创粉丝点击