Loading omniglot in few shot setting

1

what is a way to load omniglot in few shot setting, like 1 shot 5 way, or 5 shot 5 way?