Improve efficiency of backprop of padding variable

This commit is contained in:
Matthew Honnibal 2017-11-03 00:49:11 +01:00
parent a22f96c3f1
commit 260e6ee3fb

View File

@ -150,10 +150,8 @@ class PrecomputableAffine(Model):
def _backprop_padding(self, dY, ids):
# (1, nF, nO, nP) += (nN, nF, nO, nP) where IDs (nN, nF) < 0
for i in range(ids.shape[0]):
for j in range(ids.shape[1]):
if ids[i,j] < 0:
self.d_pad[0,j] += dY[i, j]
d_pad = dY * (ids.reshape((ids.shape[0], self.nF, 1, 1)) < 0.)
self.d_pad += d_pad.sum(axis=0)
return dY, ids
@staticmethod