Import Library¶In [1]:import torchimport torch.nn as nnimport torchvisionimport torchvision.transforms as transformsimport numpy as npimport mathimport matplotlib.pyplot as pltimport PILimport urllibdevice = torch.device("cuda:0")U-Net¶In [2]:class SinusoidalPositionEmbeddings(nn.Module): # position embedding def __init__(self, dim): super().__init__() self.dim = dim def for..