Working with IPython.core.display.Image

The error was generated when using Resize transform from torchvision