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