|
|
|
@ -76,8 +76,7 @@ def db_lstm(word, predicate, ctx_n2, ctx_n1, ctx_0, ctx_p1, ctx_p2, mark,
|
|
|
|
|
emb_layers.append(mark_embedding)
|
|
|
|
|
|
|
|
|
|
hidden_0_layers = [
|
|
|
|
|
fluid.layers.fc(input=emb, size=hidden_dim, act='tanh')
|
|
|
|
|
for emb in emb_layers
|
|
|
|
|
fluid.layers.fc(input=emb, size=hidden_dim) for emb in emb_layers
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
hidden_0 = fluid.layers.sums(input=hidden_0_layers)
|
|
|
|
@ -94,8 +93,8 @@ def db_lstm(word, predicate, ctx_n2, ctx_n1, ctx_0, ctx_p1, ctx_p2, mark,
|
|
|
|
|
|
|
|
|
|
for i in range(1, depth):
|
|
|
|
|
mix_hidden = fluid.layers.sums(input=[
|
|
|
|
|
fluid.layers.fc(input=input_tmp[0], size=hidden_dim, act='tanh'),
|
|
|
|
|
fluid.layers.fc(input=input_tmp[1], size=hidden_dim, act='tanh')
|
|
|
|
|
fluid.layers.fc(input=input_tmp[0], size=hidden_dim),
|
|
|
|
|
fluid.layers.fc(input=input_tmp[1], size=hidden_dim)
|
|
|
|
|
])
|
|
|
|
|
|
|
|
|
|
lstm = fluid.layers.dynamic_lstm(
|
|
|
|
|