ba03571f28ffb877b16643304494c060c0dd21f4,se3cnn/SO3.py,,spherical_harmonics,#,110

Before Change

    Js = Js.reshape(-1, *[1] * alpha.dim())
    Ms = Ms.reshape(-1, *[1] * alpha.dim())
    alpha = alpha.view(1, *alpha.size())
    beta = beta.view(1, *beta.size())
    Y = sh(Js, Ms, math.pi - beta.cpu().numpy(), alpha.cpu().numpy())
    return torch.tensor(Y, dtype=dtype, device=device)

After Change

    Ms = np.concatenate([np.arange(-J, J + 1, 1) for J in order], 0)
    Js = Js.reshape(-1, *[1] * alpha.dim())
    Ms = Ms.reshape(-1, *[1] * alpha.dim())
    alpha = alpha.unsqueeze(0)
    beta = beta.unsqueeze(0)
    Y = sh(Js, Ms, math.pi - beta.cpu().numpy(), alpha.cpu().numpy())
    return torch.tensor(Y, dtype=dtype, device=device)

Instances