From ad5a03c6f60e5b2d543326bf8917b48e5b390b82 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=C3=89tienne=20Simon?= Date: Sun, 21 Jun 2015 17:26:03 -0400 Subject: Forgot a transpose in memory network --- model/memory_network.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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( -- cgit v1.2.3