5ad7a33a04d8829ad3439b5f9390bd136105f986,allennlp/nn/beam_search.py,BeamSearch,search,#BeamSearch#Any#Any#Any#,48

Before Change

            reconstructed_predictions.append(cur_preds)

            // shape: (batch_size, beam_size)
            cur_backpointers = backpointers[timestep - 1].gather(1, cur_backpointers)

        // shape: (batch_size, beam_size, 1)
        final_preds = predictions[0].gather(1, cur_backpointers).unsqueeze(2)

After Change

        from inspect import signature

        step_signature = signature(step)
        if len(step_signature.parameters) < 3:
            old_step = cast(StepFunctionTypeNoTimestep, step)

            def new_step(
                last_predictions: torch.Tensor, state: Dict[str, torch.Tensor], time_step: int
            ):
                return old_step(last_predictions, state)

            step = new_step

        batch_size = start_predictions.size()[0]

        // List of (batch_size, beam_size) tensors. One for each time step. Does not
        // include the start symbols, which are implicit.

Instances