x.index_add_(dim, index, source, *, alpha=1)
x = index_add(dim, index, source, *, alpha=1)x の index[i] 番目に source[i] の alpha 倍を足す関数。
self[index[i], :, :] += alpha * src[i, :, :] # if dim == 0
self[:, index[i], :] += alpha * src[:, i, :] # if dim == 1
self[:, :, index[i]] += alpha * src[:, :, i] # if dim == 2この計算式のインデクシングのパターンは torch.gather と同じ。