Función PyTorch torch.scatter
Manual de referencia de Pytorch torch
torch.scatterEs una función de PyTorch que se utiliza para dispersar valores en posiciones especificadas.
Definición de la función
torch.scatter(input, dim, index, src, reduce)
Ejemplo de uso
Ejemplo
import torch
x = torch.zeros(3, 4)
# Dispersar los valores en las posiciones especificadas
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)
# Dispersar los valores en las posiciones especificadas
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)
Otras extensiones