import torch


def _admm_aux(y, sett):
    """ Get ADMM variables z and w.

    Returns:
        z (torch.tensor()): (C, 3, dim_y)
        w (torch.tensor()): (C, 3, dim_y)

    """
    # Parse function settings/parameters
    dim_y = y[0].dim
    dim = (len(y), 3) + dim_y
    # Allocate
    z = torch.zeros(dim, dtype=torch.float32, device=sett.device)
    w = torch.zeros(dim, dtype=torch.float32, device=sett.device)

    return z, w