Exemple #1
0
 def src_token_mask(self):
     return nn_utils.length_array_to_mask_tensor(self.src_sents_len,
                                                 cuda=self.cuda)
Exemple #2
0
 def value_token_mask(self):
     return nn_utils.length_array_to_mask_tensor(self.value_len,
                                                 cuda=self.cuda)
Exemple #3
0
 def table_unk_mask(self):
     return nn_utils.length_array_to_mask_tensor(self.col_len,
                                                 cuda=self.cuda,
                                                 value=None)
Exemple #4
0
 def table_token_mask(self):
     return nn_utils.length_array_to_mask_tensor(self.col_len,
                                                 cuda=self.cuda)