diffusion_models.models.nn.conv_nd

diffusion_models.models.nn.conv_nd(dims, *args, **kwargs)

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