Learning feed-forward one-shot learners