Building a One-shot Learning Network with PyTorch