diff options
Diffstat (limited to 'model/memory_network.py')
| -rw-r--r-- | model/memory_network.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/model/memory_network.py b/model/memory_network.py index c17f9b3..92e83e2 100644 --- a/model/memory_network.py +++ b/model/memory_network.py @@ -52,7 +52,7 @@ class Model(Initializable): candidate_inputs = tensor.concatenate(candidate_extremities + candidate_embeddings, axis=1) candidate_representation = self.candidate_encoder.apply(candidate_inputs) - similarity_score = tensor.dot(prefix_representation, candidate_representation) + similarity_score = tensor.dot(prefix_representation, candidate_representation.T) similarity = self.softmax.apply(similarity_score) candidate_destination = tensor.concatenate( |