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