Fonction torch.scatter de PyTorch
Manuel de référence PyTorch torch
torch.scatterIl s'agit d'une fonction de PyTorch utilisée pour disperser des valeurs vers des positions spécifiées.
Définition de fonction
torch.scatter(input, dim, index, src, reduce)
Exemple d'utilisation
Exemple
import torch
x = torch.zeros(3, 4)
# Répartir les valeurs vers des positions spécifiées
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)
# Répartir les valeurs vers des positions spécifiées
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)
Autres extensions