d22d15bde4ed4b5e5077f43a49888fd1dba313ed,liegroups/torch.py,SO2,vee,#Any#Any#,154

Before Change

    @classmethod
    def vee(cls, Phi):
        if Phi.dim() < 3:
            Phi.unsqueeze_(dim=0)

        if Phi.shape[1:3] != (cls.dim, cls.dim):
            raise ValueError(
                "Phi must have shape ({},{}) or (N,{},{})".format(cls.dim, cls.dim, cls.dim, cls.dim))

After Change

    @classmethod
    def vee(cls, Phi):
        if Phi.dim() < 3:
            Phi = Phi.unsqueeze(dim=0)

        if Phi.shape[1:3] != (cls.dim, cls.dim):
            raise ValueError(
                "Phi must have shape ({},{}) or (N,{},{})".format(cls.dim, cls.dim, cls.dim, cls.dim))

Instances