脚本专栏 
首页 > 脚本专栏 > 浏览文章

TensorFlow 合并/连接数组的方法

(编辑:jimmy 日期: 2025/7/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 合并/连接数组的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。

上一篇:python+influxdb+shell编写区域网络状况表
下一篇:pytorch训练imagenet分类的方法