  The UNet2DModel forward method.  UNet2DOutput  class diffusers.models.unet_2d.UNet2DOutput  < source > ( sample: FloatTensor  )   Parameters   sample (torch.FloatTensor of shape (batch_size, num_channels, height, width)) —