def collater()

in src/run_paraphrase.py [0:0]

• 8 lines of code
• 6 McCabe index (conditional complexity)

def collater(batch):
    batch = list(batch)
    x1 = collate_tokens([x[0] for x in batch], pad_idx=PAD_TOKEN)
    x2 = collate_tokens([x[1] for x in batch], pad_idx=PAD_TOKEN)
    len1 = torch.stack([x[2] for x in batch])
    len2 = torch.stack([x[3] for x in batch])
    y = torch.stack([x[4] for x in batch])
    return (x1, x2, len1, len2, y)