PyTorch torch.scatter 関数
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)
その他の拡張