8a5b6efbc8fdf1c63f3633c344b4aacd53e3bc52,s2cnn/ops/so3_localft.py,,setup_so3_local_ft,#Any#Any#Any#,42

Before Change

    F = F.view("float")

    // convert to torch Tensor
    F = like.new_tensor(F.astype(np.float32))

    return F

After Change

    F = F.view("float")

    // convert to torch Tensor
    F = torch.tensor(F.astype(np.float32), dtype=torch.float32, device=device)  // pylint: disable=E1102

    return F

Instances