PyTorch torch.scatter 函数

PyTorch torch.scatter 函数 Pytorch torch 参考手册 torch.scatter 是 PyTorch 中用于将值散布到指定位置的函数。函数定义 torch.scatter(input, dim, index, src, reduce) 使用示例实例 [mycode4 type='python'] import torch x = torch.zeros(3,..

Updated 7/16/2026

Tutorial detail pages default to noindex.

PyTorch torch.scatter 函数

Pytorch torch 参考手册

torch.scatter 是 PyTorch 中用于将值散布到指定位置的函数。

函数定义

torch.scatter(input, dim, index, src, reduce)

使用示例

实例

import torch

x = torch.zeros(3, 4)

# 将值散布到指定位置
index = torch.tensor([[0], [1], [2]])
src = torch.tensor([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])

result = torch.scatter(x, dim=0, index=index, src=src)

print(result)

Pytorch torch 参考手册