如何正确安装torch_scatter

要安装torch_scatter，可以按照以下步骤进行：

1. 确保已经安装了PyTorch和CUDA（如果需要）。

2. 打开终端或命令行，并使用以下命令安装torch_scatter：

   pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.9.0+cu111.html

注意，这将安装适用于PyTorch 1.9.0和CUDA 11.1的torch_scatter版本。如果您使用的是不同的PyTorch或CUDA版本，请访问 https://pytorch-geometric.com/ 中的“Installation”部分，查找适合您的版本的安装命令。

3. 安装完成后，您可以在Python中导入torch_scatter并使用它。例如：

   import torch
   from torch_scatter import scatter_mean
   
   x = torch.randn(4, 5)
   row = torch.tensor([0, 0, 1, 2])
   out = scatter_mean(x, row, dim=0)
   
   print(out)

这将对x中每一行进行平均汇总，并将结果存储在out中。