851b618c26a3ab02076e7f40a54d7542e404013b,se3cnn/SO3.py,,spherical_harmonics_xyz,#,109

Before Change

def spherical_harmonics_xyz(x, y, z, J):
    with torch_default_dtype(torch.float64):
        if x == y == z == 0:  // angles at origin are nan, special treatment
            if J == 0:  // Y^0 is angularly independent, choose any angle
                return spherical_harmonics(0, 123, 321)  // [m]
            else:  // insert zeros for Y^J with J!=0
                return 0
        else:  // not at the origin, sample spherical harmonic
            alpha, beta = x_to_alpha_beta([x, y, z])
            return spherical_harmonics(J, alpha, beta)  // [m]

After Change

        // fix values when xyz = 0
        val = torch.cat([spherical_harmonics(0, 123, 321) if J == 0 else torch.zeros(2 * J + 1) for J in order])  // [m]
        out[:, xyz.norm(2, -1) == 0] = val.view(-1, 1)
        return out


def compose(a1, b1, c1, a2, b2, c2):

Instances