PyTorch torch.scatter-Funktion
PyTorch torch-Referenzhandbuch
torch.scatterEs ist eine Funktion in PyTorch, die verwendet wird, um Werte an angegebene Positionen zu streuen.
Funktionsdefinition
torch.scatter(input, dim, index, src, reduce)
Anwendungsbeispiel
Beispiel
import torch
x = torch.zeros(3, 4)
# Streue Werte an angegebene Positionen
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)
# Streue Werte an angegebene Positionen
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)
Andere Erweiterungen