def get_prior_z(device):
    ' Returns prior distribution for latent code z.\n    Args:\n        cfg (dict): imported yaml config\n        device (device): pytorch device\n    '
    z_dim = 0
    p0_z = dist.Normal(torch.zeros(z_dim, device=device), torch.ones(z_dim, device=device))
    return p0_z
Returns prior distribution for latent code z.
Args:
    cfg (dict): imported yaml config
    device (device): pytorch device
def get_prior_z(device):
    ' Returns prior distribution for latent code z.\n    Args:\n        cfg (dict): imported yaml config\n        device (device): pytorch device\n    '
    z_dim = 0
    p0_z = dist.Normal(torch.zeros(z_dim, device=device), torch.ones(z_dim, device=device))
    return p0_z