欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

TensorFlow 合并/连接数组的方法

程序员文章站 2023-09-04 16:10:10
如下所示: import tensorflow as tf a = tf.Variable([4,5,6]) b = tf.Variable([1,2,3...

如下所示:

import tensorflow as tf

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

c = tf.concat(0,[a,b])

init_op = tf.initialize_all_variables()

with tf.Session() as sess:
 sess.run(init_op)
 print(sess.run(c))

结果打印:

[4 5 6 1 2 3]

以上这篇TensorFlow 合并/连接数组的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。