tensorflow

tensorflow矩阵拼接的方法是什么

小亿
94
2024-03-13 20:46:28
栏目: 深度学习

在TensorFlow中,可以使用tf.concat()函数将两个或多个张量沿指定维度进行拼接。具体语法如下:

tf.concat(
    values,
    axis,
    name='concat'
)

参数说明:

示例代码:

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。

0
看了该问题的人还看了