class_docstr: Helper object to call indexed update functions for an (advanced) index.
  
  This object references a source array and a specific indexer into that array.
  Methods on this object return copies of the source array that have been
  modified at the positions specified by the indexer.
class_docstr_tok: ["Helper","object","to","call","indexed","update","functions","for","an","(","advanced",")","index",".","This","object","references","a","source","array","and","a","specific","indexer","into","that","array",".","Methods","on","this","object","return","copies","of","the","source","array","that","have","been","modified","at","the","positions","specified","by","the","indexer","."]
docstr: Pure equivalent of ``x[idx] = y``.
  
  Returns the value of ``x`` that would result from the NumPy-style
  :mod:`indexed assignment <numpy.doc.indexing>` ``x[idx] = y``.
  
  See :mod:`jax.ops` for details.
docstr_tok: ["Pure","equivalent","of","``","x","[","idx","]","=","y","``",".","Returns","the","value","of","``","x","``","that","would","result","from","the","NumPy-style",":","mod",":","`","indexed","assignment","<","numpy.doc.indexing",">","`","``","x","[","idx","]","=","y","``",".","See",":","mod",":","`","jax.ops","`","for","details","."]
code: def set(
      self,
      values,
      *,
      indices_are_sorted=False,
      unique_indices=False,
      mode=None,
  ):
      """Pure equivalent of ``x[idx] = y``.
  
      Returns the value of ``x`` that would result from the NumPy-style
      :mod:`indexed assignment <numpy.doc.indexing>` ``x[idx] = y``.
  
      See :mod:`jax.ops` for details.
      """
      return scatter._scatter_update(
          self.array,
          self.index,
          values,
          lax.scatter,
          indices_are_sorted=indices_are_sorted,
          unique_indices=unique_indices,
          mode=mode,
      )
  
imports: from __future__ import annotations
  import abc
  from functools import partial, wraps
  import math
  from typing import Any
  import numpy
  import jax
  from jax import lax
  from jax._src import core
  from jax._src import dtypes
  from jax._src.api_util import _ensure_index_tuple
  from jax._src.array import ArrayImpl
  from jax._src.lax import lax
  from jax._src.numpy import lax_numpy
  from jax._src.numpy import reductions
  from jax._src.numpy import ufuncs
  from jax._src.numpy import util
  from jax._src.ops import scatter
  from jax._src.typing import Array, ArrayLike, DimSize, DTypeLike, Shape
  from jax._src.util import safe_zip, safe_map
label_desc: Use image node_id 4 for calling the _IndexUpdateRef obj's underlying member method code with example usage: obj.set(values) and returns: scatter