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) —