溫馨提示×

tensorflow矩陣拼接的方法是什么

小億
105
2024-03-13 20:46:28
欄目: 深度學習

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

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女