PyTorch torch.scatter Function
PyTorch torch Reference Manual
torch.scatterIt is a function in PyTorch used to scatter values to specified positions.
Function Definition
torch.scatter(input, dim, index, src, reduce)
Usage Example
Example
import torch
x = torch.zeros(3, 4)
# Scatter values to specified positions
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)
x = torch.zeros(3, 4)
# Scatter values to specified positions
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)
Other Extensions