071fb5fb20683946a33f40d28af466f8e9ffc1bd,liegroups/torch/so2.py,SO2,left_jacobian,#,41

Before Change

            s = phi[large_angle_inds].sin()
            c = phi[large_angle_inds].cos()

            A = s / phi[large_angle_inds]
            B = (1. - c) / phi[large_angle_inds]

            jac_large_angle = phi.__class__(
                len(large_angle_inds), cls.dim, cls.dim)

After Change

        if len(large_angle_inds) > 0:
            angle = phi[large_angle_inds]
            s = angle.sin()
            c = angle.cos()

            A = (s / angle).unsqueeze_(dim=1).unsqueeze_(
                dim=2).expand_as(jac[large_angle_inds]) * \
                torch.eye(cls.dim).unsqueeze_(dim=0).expand_as(
                jac[large_angle_inds])
            B = ((1. - c) / angle).unsqueeze_(dim=1).unsqueeze_(
                dim=2).expand_as(jac[large_angle_inds]) * \
                cls.wedge(phi.__class__([1.]))

            jac[large_angle_inds] = A + B

        return jac.squeeze_()

    @classmethod

Instances