Function linear

sat/sgm/modules/diffusionmodules/util.py:261–265

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