scatter_add()函数中的参量设置
时间: 2023-06-01 12:04:39 浏览: 177
详解pandas绘制矩阵散点图(scatter_matrix)的方法
5星 · 资源好评率100%
scatter_add()函数的参数设置如下:
```python
torch.scatter_add(input, dim, index, src, out=None)
```
- input (Tensor):输入张量
- dim (int):指定哪个维度进行scatter操作
- index (LongTensor):指定scatter操作的索引
- src (Tensor):指定要添加的张量
- out (Tensor, optional):输出张量,如果不为None,结果将会被写入这个张量
其中,input、src和out的维度必须一致,而index的维度必须与input的维度相同,除了指定的dim维度。
阅读全文