Fix parser test

This commit is contained in:
Matthew Honnibal 2017-11-04 00:23:45 +01:00
parent 98c29b7912
commit e4ec4be948
1 changed files with 1 additions and 1 deletions

View File

@ -69,7 +69,7 @@ def test_PrecomputableAffine(nO=4, nI=5, nF=3, nP=2):
Y, get_dX = model.begin_update(tensor)
assert Y.shape == (tensor.shape[0]+1, nF, nO, nP)
assert model.d_pad.shape == (1, nF, nO, nP)
dY = model.ops.allocate((15, nF, nO, nP))
dY = model.ops.allocate((15, nO, nP))
ids = model.ops.allocate((15, nF))
ids[1,2] = -1
dY[1,2] = 1