función de concatenación de keras

Concatenar se puede ver desde el nombre para conectar dos matrices, el siguiente es un pequeño ejemplo para demostrar su uso

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


 

Supongo que te gusta

Origin blog.csdn.net/devil_son1234/article/details/107406805
Recomendado
Clasificación