Function build_scaling_rotation_4d

utils/general_utils.py:135–145

(s, l, r)

    return A

def build_scaling_rotation_4d(s, l, r):
    L = torch.zeros((s.shape[0], 4, 4), dtype=torch.float, device="cuda")
    R = build_rotation_4d(l, r)

    L[:,0,0] = s[:,0]
    L[:,1,1] = s[:,1]
    L[:,2,2] = s[:,2]
    L[:,3,3] = s[:,3]

    L = R @ L
    return L

def safe_state(silent):
    old_f = sys.stdout