Function _scatter_add

tests/python/pytorch/sparse/test_sparse_matrix.py:206–210

(a, index, v=1)

def _scatter_add(a, index, v=1):
    index = index.tolist()
    for i in index:
        a[i] += v
    return a


@pytest.mark.parametrize("dense_dim", [None, 4])