or) —
The noisy input tensor with the following shape (batch_size, num_channels, sample_size).   timestep (torch.FloatTensor or float or int) — The number of timesteps to denoise an input.   return_dict (bool, optional, defaults to True) —
Whether or not to return a UNet1DOutput instead of a plain tuple.   Returns 
UNet1DOutput or tuple
  
If return_dict is True, an UNet1DOutput is returned, otherwise a tuple is
returned where the first element