lunny/keras
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #972 from dbonadiman/patch-1

Browse Source 
[Patch] Fixing issue with dot output shape
This commit is contained in:
François Chollet 2015-11-08 18:05:35 -08:00
commit b1bbedfb46
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
keras/layers/core.py
View File

@ -406,6 +406,9 @@ class Merge(Layer):
l1 = self.layers[0].get_output(train)
l2 = self.layers[1].get_output(train)
output = T.batched_tensordot(l1, l2, self.dot_axes)
output_shape = list(self.output_shape)
output_shape[0] = l1.shape[0]
output = output.reshape(tuple(output_shape))
return output
elif self.mode == 'cos':
l1 = self.layers[0].get_output(train)