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

Before Change

        // Reconstruct the sequences.
        // shape: [(batch_size, beam_size, 1)]
        reconstructed_predictions = [predictions[-1].unsqueeze(2)]

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

After Change

            // Keep only the pieces of the state tensors corresponding to the
            // ancestors created this iteration.
            for key, state_tensor in state.items():
                if state_tensor is None:
                    continue
                _, *last_dims = state_tensor.size()
                // shape: (batch_size, beam_size, *)
                expanded_backpointer = backpointer.view(
                    batch_size, self.beam_size, *([1] * len(last_dims))

Instances