gpt4 book ai didi

merge - 凯拉斯 : How to merge a dense layer and an embedding layer

转载 作者:行者123 更新时间:2023-12-01 08:14:46 25 4
gpt4 key购买 nike

我使用 Keras 并尝试将两个不同的层连接成一个向量(向量的第一个值是第一层的值,另一部分是第二层的值)。
其中一层是密集层,另一层是嵌入层。

我知道如何合并两个嵌入层或两个密集层,但我不知道如何合并一个嵌入层和一个密集层(维度问题)。

一个简单的例子是这样的:

L_branch = Sequential()
L_branch.add(Dense(10, input_shape = (4,) , activation = 'relu'))
L_branch.add(BatchNormalization())

R_branch = Sequential()
R_branch.add(Embedding(1000, 64, input_length=5))

final_branch.add(Merge([L_branch, R_branch], mode = 'concat'))

但这不起作用,因为您无法合并具有不同维度的图层。

PS:对不起,英语不是我的母语,我希望你能理解我的问题。

此致。

最佳答案

使用 Flatten层。

L_branch = Sequential()
L_branch.add(Dense(10, input_shape = (4,) , activation = 'relu'))
L_branch.add(BatchNormalization())

R_branch = Sequential()
R_branch.add(Embedding(1000, 64, input_length=5))
R_branch.add(Flatten()) # <--

final_branch = Sequential() # <--
final_branch.add(Merge([L_branch, R_branch], mode = 'concat'))

关于merge - 凯拉斯 : How to merge a dense layer and an embedding layer,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41933830/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com