26421ce20c6b626ceacafbb3282cad1d5dce04ca,onmt/Models.py,Embeddings,forward,#Embeddings#Any#,60

Before Change

        Return:
            emb (FloatTensor): len x batch x input_size
        
        word = self.word_lut(src_input[:, :, 0])
        emb = word
        if self.feature_dicts:
            features = [feature_lut(src_input[:, :, j+1])

After Change

        Return:
            emb (FloatTensor): len x batch x sum of feature embedding sizes
        
        feat_inputs = (feat.squeeze(2) for feat in src_input.split(1, dim=2))
        features = [lut(feat) for lut, feat in zip(self.emb_luts, feat_inputs)]
        emb = self.merge(features)
        return emb


class Encoder(nn.Module):

Instances