keras concatenate函数

最新推荐文章于 2025-11-05 13:08:29 发布

原创最新推荐文章于 2025-11-05 13:08:29 发布 · 3.1w 阅读

9 ·

CC 4.0 BY-SA版权

keras 专栏收录该内容

5 篇文章

订阅专栏

本文通过一个简单示例展示了如何使用 Keras 的 Concatenate 函数将两个数组沿指定轴进行合并。示例中使用了 TensorFlow 和 NumPy 进行变量初始化，并通过 Keras 后端实现了两个一维数组的拼接。

部署运行你感兴趣的模型镜像

concatenate从名字可以看出是连接两个数组，下面用一个小例子演示其用法

import numpy as np
import cv2
import keras.backend as K
import tensorflow as tf


a = K.variable(np.array([[1 , 2, 3]]))
b = K.variable(np.array([[3 , 2, 1]]))
c1 = K.concatenate([a , b] , axis=0)
c2 = K.concatenate([a , b] , axis=-1)

init = tf.global_variables_initializer()
with tf.Session() as sess:
    sess.run(init)
    print(sess.run(c1))
    print(sess.run(c2))

# [[1. 2. 3.]
#  [3. 2. 1.]]
# [[1. 2. 3. 3. 2. 1.]]