keras concatenate函数

最新推荐文章于 2021-10-03 15:57:54 发布

转载最新推荐文章于 2021-10-03 15:57:54 发布 · 1k 阅读

1 ·

CC 4.0 BY-SA版权

原文链接：https://blog.youkuaiyun.com/hanshuobest/article/details/79488209

文章标签：

#python

语法专栏收录该内容

50 篇文章

订阅专栏

本文通过实例演示了如何使用Keras的Concatenate函数来连接两个数组。详细解释了axis参数的作用，展示了在不同axis设置下，数组是如何被连接的。

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.]]