ae2d6eb40138e9fe09d65469c4b1c4a0fc4027c6,se3cnn/SO3.py,,spherical_harmonics_xyz,#Any#Any#Any#Any#,276

Before Change

    :param xyz: tensor of shape [..., 3]
    :return: tensor of shape [m, ...]
    
    if not isinstance(order, list):
        order = [order]

    with torch_default_dtype(torch.float64):

After Change

    except TypeError:
        order = [order]

    if dtype is None and torch.is_tensor(xyz):
        dtype = xyz.dtype
    if dtype is None:
        dtype = torch.get_default_dtype()

    if device is None and torch.is_tensor(xyz):
        device = xyz.device

    if not torch.is_tensor(xyz):
        xyz = torch.tensor(xyz, dtype=torch.float64)

Instances