Appends dimensions to the end of a tensor until it has target_dims dimensions.

Create a linear module.

Create a 1D, 2D, or 3D convolution module.

Evaluate a function without caching intermediate activations, allowing for reduced memory at the expense of extra compute in the backward pas

Create sinusoidal timestep embeddings. :param timesteps: a 1-D Tensor of N indices, one per batch element. These may be

Make a standard normalization layer. :param channels: number of input channels. :return: an nn.Module for normalization.

Zero out the parameters of a module and return it.