def init_feedback_indices_qg(q, g, positive=False, device=None):
    indices = torch.zeros(q, (q + g), dtype=torch.bool, device=device)
    if positive:
        indices[(torch.arange(q), torch.arange(q))] = True
    return indices
def init_feedback_indices_qg(q, g, positive=False, device=None):
    indices = torch.zeros(q, (q + g), dtype=torch.bool, device=device)
    if positive:
        indices[(torch.arange(q), torch.arange(q))] = True
    return indices