f66d3c5ed7b7fa96424fd0d8214594042d9ba3ff,se3cnn/SO3.py,,irr_repr,#,81

Before Change

        beta = beta.item()
    if torch.is_tensor(gamma):
        gamma = gamma.item()
    return torch.tensor(wigner_D_matrix(order, alpha, beta, gamma), dtype=torch.get_default_dtype() if dtype is None else dtype)



////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

After Change

            if dtype is None:
                dtype = x.dtype
            if device is None:
                device = x.device
    if dtype is None:
        dtype = torch.get_default_dtype()
    return torch.tensor(wigner_D_matrix(order, *abc), dtype=dtype, device=device)



////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

Instances