import torch as tmatplotlib inlinefrom matplotlib import pyplot as pltfrom Ipython import display
It seems like there is a typo in your code. The correct import statement for IPython should be:
from IPython import display
Please update the import statement and try again.