Added cifar generator

cc @eickenberg 
`np.newaxis`