UNet2DOutput  
The output of [`UNet2DModel`].  
Args:
sample (`torch.Tensor` of shape `(batch_size, num_channels, height, width)`):
The hidden states output from the last layer of the model.
/Users/nielsrogge/Documents/python_projecten/diffusers/docs/source/en/api/models/unet2d.md