Function linear

diff2flow/models/unet/util.py:128–132

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