what is causing this error ?
add_elastic_transform(image, alpha, sigma, pad_size, seed)
     29 
     30     x, y = np.meshgrid(np.arange(shape[1]), np.arange(shape[0]))
---> 31     indices = np.reshape(y+dy, (-1, 1)), np.reshape(x+dx, (-1, 1))
     32     return cropping(map_coordinates(image, indices, order=1).reshape(shape), 512, pad_size, pad_size), seed
    

is it because of RGB channels ?