diffusion_models.models.nn.avg_pool_nd

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

Create a 1D, 2D, or 3D average pooling module.