在TensorFlow中,可以使用tf.concat()
函数将两个或多个张量沿指定维度进行拼接。具体语法如下:
tf.concat(
values,
axis,
name='concat'
)
参数说明:
values
:要拼接的张量列表,可以是一个包含张量的列表。axis
:指定拼接的维度。name
:可选参数,操作的名称。示例代码:
import tensorflow as tf
# 创建两个张量
A = tf.constant([[1, 2], [3, 4]])
B = tf.constant([[5, 6]])
# 在axis=0维度上拼接
C = tf.concat([A, B], axis=0)
with tf.Session() as sess:
result = sess.run(C)
print(result)
输出结果:
[[1 2]
[3 4]
[5 6]]
在上面的示例中,我们将两个2D张量A和B沿着axis=0维度进行拼接,得到了一个新的2D张量C。