INNER CODE UNIT · Python

mask

IBM/transition-amr-parser · src/ibm_neural_aligner/main.py:637

                mask = mask + [MaskInfo.unchanged] * edge_size

            m = torch.tensor(mask, dtype=torch.long)
            batch_mask_gcn.append(m)
        batch_map['mask'] = batch_mask
        batch_map['mask_for_gcn'] = batch_mask_gcn

    return batch_map


class Embed(nn.Module):
    def __init__(self, vocab_size, size, project_size=0, padding_idx=PADDING_IDX, embed=None, dropout_p=0, mode='learn'):
        super().__init__()

        self.mode = mode
        if mode == 'learn':
            self.embed = nn.Embedding(vocab_size, size, padding_idx=padding_idx)
            self.esize = self.embed.weight.shape[1]

View source record →