loop instead of any
@@ -291,14 +291,16 @@ def forward(model: Tok2VecListener, inputs, is_train: bool):
# of data.
# When the components batch differently, we don't receive a matching
# prediction from the upstream, so we can't predict.
- if not any(doc.tensor.size for doc in inputs):
+ outputs = []
+ width = model.get_dim("nO")
+ for doc in inputs:
+ if doc.tensor.size == 0:
# But we do need to do *something* if the tensor hasn't been set.
# The compromise is to at least return data of the right shape,
# so the output is valid.
- width = model.get_dim("nO")
- outputs = [model.ops.alloc2f(len(doc), width) for doc in inputs]
+ outputs.append(model.ops.alloc2f(len(doc), width))
else:
- outputs = [doc.tensor for doc in inputs]
+ outputs.append(doc.tensor)
return outputs, lambda dX: []
