2d.UNet2DOutput

<
source
>
(
sample: FloatTensor

)


Parameters 

sample (torch.FloatTensor of shape (batch_size, num_channels, height, width)) —
Hidden states output. Output of last layer of model.




UNet2DModel
	

class diffusers.UNet2DModel

<
source
>
(
sample_size: typing.Union[int, typing.Tuple[int, int], NoneType] = None
in_channels: int = 3
out_channels: int = 3
center_input_sample: bool = False
time_embedding_type: str = 'positiona