def compute_b_error(b: torch.tensor, bhat: torch.tensor) -> float: """ compute l2 norm of bhat (unit-normalized) - b (unit-normalized) """ b_normalized = b.div(torch.norm(b, p=2)) bhat_normalized = bhat.div(torch.norm(bhat, p=2)) return torch.norm(b_normalized - bhat_normalized, p=2).item()
def normalize(t: torch.tensor, p=2) -> torch.tensor: """ make unit norm w.r.t l-p norm """ return t.div(torch.norm(t, p=p))