**Please Write Function in Python**

b. Write a function that can shift an MNIST image in any direction (left, right, up, or down) by one pixel Then, for each image in the training set, create four shifted copies (one per direction) and add them to the training set. Finally, train your best model on this expanded training set and measure its accuracy on the test set. You can use the shift) function from the scipy.ndimage.interpolation module. For example, shift image, [2, 1], cval-0) shifts the image 2 pixels down and 1 pixel to the right