本文简要介绍python语言中 torch.Tensor.scatter_add_
的用法。
用法:
Tensor.scatter_add_(dim, index, src) → Tensor
以与
scatter_()
类似的方式将来自张量other
的所有值添加到self
在index
张量中指定的索引处。对于src
中的每个值,将其添加到self
中的索引中,该索引由src
中的索引指定dimension != dim
和index
中的相应值dimension = dim
。对于 3-D 张量,
self
更新为:self[index[i][j][k]][j][k] += src[i][j][k] # if dim == 0 self[i][index[i][j][k]][k] += src[i][j][k] # if dim == 1 self[i][j][index[i][j][k]] += src[i][j][k] # if dim == 2
self
、index
和src
应该具有相同的维数。还需要index.size(d) <= src.size(d)
用于所有维度d
,并且index.size(d) <= self.size(d)
用于所有维度d != dim
。请注意,index
和src
不广播。注意
当给定 CUDA 设备上的张量时,此操作可能会表现得不确定。有关详细信息,请参阅重现性。
注意
向后传递仅针对
src.shape == index.shape
实施。例子:
>>> src = torch.ones((2, 5)) >>> index = torch.tensor([[0, 1, 2, 0, 0]]) >>> torch.zeros(3, 5, dtype=src.dtype).scatter_add_(0, index, src) tensor([[1., 0., 0., 1., 1.], [0., 1., 0., 0., 0.], [0., 0., 1., 0., 0.]]) >>> index = torch.tensor([[0, 1, 2, 0, 0], [0, 1, 2, 2, 2]]) >>> torch.zeros(3, 5, dtype=src.dtype).scatter_add_(0, index, src) tensor([[2., 0., 0., 1., 1.], [0., 2., 0., 0., 0.], [0., 0., 2., 1., 1.]])
参数:
相关用法
- Python PyTorch Tensor.scatter_用法及代码示例
- Python PyTorch Tensor.storage_offset用法及代码示例
- Python PyTorch Tensor.sparse_mask用法及代码示例
- Python PyTorch Tensor.stride用法及代码示例
- Python PyTorch Tensor.size用法及代码示例
- Python PyTorch Tensor.unflatten用法及代码示例
- Python PyTorch Tensor.register_hook用法及代码示例
- Python PyTorch Tensor.to用法及代码示例
- Python PyTorch Tensor.is_leaf用法及代码示例
- Python PyTorch Tensor.imag用法及代码示例
- Python PyTorch Tensor.unfold用法及代码示例
- Python PyTorch Tensor.real用法及代码示例
- Python PyTorch Tensor.refine_names用法及代码示例
- Python PyTorch Tensor.rename用法及代码示例
- Python PyTorch Tensor.view用法及代码示例
- Python PyTorch Tensor.new_empty用法及代码示例
- Python PyTorch Tensor.index_copy_用法及代码示例
- Python PyTorch Tensor.new_tensor用法及代码示例
- Python PyTorch Tensor.fill_diagonal_用法及代码示例
- Python PyTorch Tensor.repeat用法及代码示例
- Python PyTorch Tensor.item用法及代码示例
- Python PyTorch Tensor.tolist用法及代码示例
- Python PyTorch Tensor.put_用法及代码示例
- Python PyTorch Tensor.map_用法及代码示例
- Python PyTorch Tensor.index_fill_用法及代码示例
注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.Tensor.scatter_add_。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。