修改模型嵌入层

tensnsorflow/train.py

@@ -120,7 +120,7 @@ def model_fn(features, labels, mode, params):
     with tf.variable_scope("Shared-Embedding-layer"):
         embedding_id = tf.nn.embedding_lookup(Feat_Emb,feat_ids)
 
-        x_concat = tf.concat(tf.reshape(embedding_id,shape=[-1, common_dims],axis=1))  # None * (F * K)
+        x_concat = tf.reshape(embedding_id,shape=[-1, common_dims])  # None * (F * K)
 
     with tf.name_scope("CVR_Task"):
         if mode == tf.estimator.ModeKeys.TRAIN: