Function linear

sgm/modules/diffusionmodules/util.py:291–295

Create a linear module.

(*args, **kwargs)

def linear(*args, **kwargs):
    """
    Create a linear module.
    """
    return nn.Linear(*args, **kwargs)


def avg_pool_nd(dims, *args, **kwargs):