多gpu下Keras提取嵌入层权重

Question

在我的设置中，我训练了一个密集的keras层，其中层的权重成为图像嵌入。然后我取这个层的权重，用一组输入向量做点积。在

我想用GPU运行这个，用1个GPU就可以了。但是，我注意到，在数据并行模式下遵循多个gpu的documentation时，出现的模型摘要显示了网络结构的变化。看起来这是因为它“在每个设备上复制一次目标模型，并使用每个副本处理不同部分的输入数据”。在

问题：当我从嵌入层中提取权重时，它是否仍会提供整个层，还是需要提取多组权重并以某种方式合并它们？在