def METHOD_NAME(params, indices, axis, output):
    batch_dims = 1
    result = []
    for p, i, o in zip(params, indices, output):
        r = _scatter_add_numpy(np.ones_like(o), axis - batch_dims, i, p.shape)
        result.append(r)
    return np.stack(result)