def get_code_change_tensors(self, code_vocab: VocabEntry,
action_vocab: VocabEntry,
device: torch.device):
code_tensor_a = code_vocab.to_input_tensor(self.old_code_tokens,
device)
code_tensor_b = code_vocab.to_input_tensor(self.new_code_tokens,
device)
edit_tensor = action_vocab.to_input_tensor(self.edit_actions, device)
return code_tensor_a, code_tensor_b, edit_tensor
def get_tgt_out_tensor(self, vocab: VocabEntry,
device: torch.device) -> Tensor:
return vocab.to_input_tensor(self.tgt_out_tokens, device)
