Функция torch.scatter в PyTorch
Справочное руководство 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)
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)
Другие расширения