diff --git a/spacy/pipeline/tok2vec.py b/spacy/pipeline/tok2vec.py index f2d138cf7..a228ea815 100644 --- a/spacy/pipeline/tok2vec.py +++ b/spacy/pipeline/tok2vec.py @@ -291,8 +291,14 @@ class Tok2VecListener(Model): def forward(model: Tok2VecListener, inputs, is_train: bool): """Supply the outputs from the upstream Tok2Vec component.""" + if is_train: model.verify_inputs(inputs) return model._outputs, model._backprop else: - return [doc.tensor for doc in inputs], lambda dX: [] + if model._outputs is None: + outputs = [model.ops.alloc2f(len(doc), width) for doc in inputs] + else: + outputs = model._outputs + model._outputs = None + return outputs, lambda dX: []