|
|
|
@ -50,28 +50,18 @@ next_word = layers.data(
|
|
|
|
|
program=program,
|
|
|
|
|
init_program=init_program)
|
|
|
|
|
|
|
|
|
|
embed_param_attr_1 = {
|
|
|
|
|
'name': 'shared_w',
|
|
|
|
|
'init_attr': {
|
|
|
|
|
'max': 1.0,
|
|
|
|
|
'type': 'uniform_random',
|
|
|
|
|
'min': -1.0
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
embed_param_attr_2 = {'name': 'shared_w'}
|
|
|
|
|
|
|
|
|
|
embed_first = layers.embedding(
|
|
|
|
|
input=first_word,
|
|
|
|
|
size=[dict_size, embed_size],
|
|
|
|
|
data_type='float32',
|
|
|
|
|
param_attr=embed_param_attr_1,
|
|
|
|
|
param_attr={'name': 'shared_w'},
|
|
|
|
|
program=program,
|
|
|
|
|
init_program=init_program)
|
|
|
|
|
embed_second = layers.embedding(
|
|
|
|
|
input=second_word,
|
|
|
|
|
size=[dict_size, embed_size],
|
|
|
|
|
data_type='float32',
|
|
|
|
|
param_attr=embed_param_attr_2,
|
|
|
|
|
param_attr={'name': 'shared_w'},
|
|
|
|
|
program=program,
|
|
|
|
|
init_program=init_program)
|
|
|
|
|
|
|
|
|
@ -79,14 +69,14 @@ embed_third = layers.embedding(
|
|
|
|
|
input=third_word,
|
|
|
|
|
size=[dict_size, embed_size],
|
|
|
|
|
data_type='float32',
|
|
|
|
|
param_attr=embed_param_attr_2,
|
|
|
|
|
param_attr={'name': 'shared_w'},
|
|
|
|
|
program=program,
|
|
|
|
|
init_program=init_program)
|
|
|
|
|
embed_forth = layers.embedding(
|
|
|
|
|
input=forth_word,
|
|
|
|
|
size=[dict_size, embed_size],
|
|
|
|
|
data_type='float32',
|
|
|
|
|
param_attr=embed_param_attr_2,
|
|
|
|
|
param_attr={'name': 'shared_w'},
|
|
|
|
|
program=program,
|
|
|
|
|
init_program=init_program)
|
|
|
|
|
|
|
|
|
|