Function linear

improved_diffusion/nn.py:35–39

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