Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix tensor gradient in parser

This commit is contained in:
Matthew Honnibal 2017-08-15 03:14:36 -05:00
parent e420e0366c
commit a8e4064dd8
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
spacy/syntax/nn_parser.pyx
View File

@ -633,10 +633,9 @@ cdef class Parser:
xp = get_array_module(d_tokvecs)
for ids, d_vector, bp_vector in backprops:
d_state_features = bp_vector(d_vector, sgd=sgd)
mask = ids >= 0
indices = xp.nonzero(mask)
self.model[0].ops.scatter_add(d_tokvecs, ids[indices],
d_state_features[indices])
mask = (ids >= 0).reshape((ids.shape[0], ids.shape[1], 1))
self.model[0].ops.scatter_add(d_tokvecs, ids,
d_state_features * mask)
@property
def move_names(self):