Method squeeze

onmt/decoders/ensemble.py:23–27

Delegate squeeze to avoid modifying :func:`onmt.translate.translator.Translator.translate_batch()`

(self, dim=None)

        self.model_dec_outs = tuple(model_dec_outs)

    def squeeze(self, dim=None):
        """Delegate squeeze to avoid modifying
        :func:`onmt.translate.translator.Translator.translate_batch()`
        """
        return EnsembleDecoderOutput([x.squeeze(dim) for x in self.model_dec_outs])

    def __getitem__(self, index):
        return self.model_dec_outs[index]