4cbe889c27f5a0c272c54b4e1065ef3525c2fc09,s2cnn/nn/soft/so3_integrate.py,,setup_so3_integrate,#Any#Any#,32

Before Change

    w = S3.quadrature_weights(b) // (2b) [beta]

    w = torch.FloatTensor(w)

    if cuda_device is not None:
        w = w.cuda(cuda_device)

    return w

After Change

def setup_so3_integrate(b, like):
    import lie_learn.spaces.S3 as S3

    return like.new_tensor(S3.quadrature_weights(b)) // (2b) [beta]

Instances