如何在keras中设置嵌入层名称

1 回答

• 0

  您可以像设置任何其他图层一样设置嵌入图层的名称 .

  from keras.layers import Embedding, Input
  from keras import Model
  
  
  inputs_bedding = Input(shape=(32,)) 
  embedding = Embedding(9488, 512, trainable=False, name="test")(inputs_bedding)
  
  model = Model(inputs=inputs_bedding, outputs=embedding)
  

  model.summary（）为您提供：

  _________________________________________________________________
  Layer (type)                 Output Shape              Param #   
  =================================================================
  input_3 (InputLayer)         (None, 32)                0         
  _________________________________________________________________
  test (Embedding)             (None, 32, 512)           4857856   
  =================================================================
  Total params: 4,857,856
  Trainable params: 0
  Non-trainable params: 4,857,856
  _________________________________________________________________
  

  回复于 2024-04-29T05:12:10+08:00