Thank you , i added the Z dim too but my images is RGB and thier labels is gray scales,  when I apply add_elastic_transform it gives me error,  should I change the images to gray scale or should I add else  in add_elastic_transform for just x,y dim if so how can I add it .