fix division for segment_coo_mean when input is of type long

Author: rusty1s

csrc/cuda/segment_coo_cuda.cu

@@ -273,7 +273,10 @@ segment_coo_cuda(torch::Tensor src, torch::Tensor index,
         auto count = arg_out.value();
         for (int i = dim + 1; i < out.dim(); i++)
           count = count.unsqueeze(-1);
-        out.div_(count);
+        if (out.is_floating_point())
+          out.true_divide_(count);
+        else
+          out.floor_divide_(count);
       }
     });
   });