neric methods the library
implements for all the model (such as downloading or saving, etc.)

forward

<
source
>
(
sample: FloatTensor
timestep: typing.Union[torch.Tensor, float, int]
return_dict: bool = True

)
→
UNet1DOutput or tuple

Parameters 

sample (torch.FloatTensor) — (batch_size, num_channels, sample_size) noisy inputs tensor


timestep (torch.FloatTensor or float or `int) — (batch) timesteps


return_dict (bool, optional, defaults 