tensorflow矩阵拼接的方法是什么

寻技术 人工智能 前天 21:34 5

在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。

关闭

用微信“扫一扫”