c048078187945e1defb8abc3246cd305fb038c6a,OpenNMT/onmt/Models.py,Decoder,forward,#Decoder#Any#Any#Any#,104

Before Change

            output = self.dropout(output)
            outputs += [output]

        outputs = torch.cat(outputs)
        pred = self.logsoftmax(self.decoder(outputs))
        return pred

After Change

            output = self.dropout(output)
            outputs += [output]

        outputs = torch.cat([x.unsqueeze(0) for x in outputs])
        return outputs


class NMTModel(nn.Container):

Instances