class_docstr: A side-effect associated with the input of a jaxpr.
  
  Note that the `input_index` includes constvars.
class_docstr_tok: ["A","side-effect","associated","with","the","input","of","a","jaxpr",".","Note","that","the","`","input_index","`","includes","constvars","."]
code: def replace(self, *, input_index: Any | None = None):
      if input_index is None:
          input_index = self.input_index
      return self.__class__(input_index)
  
file_path: jax/jax/_src/effects.py
imports: from __future__ import annotations
  from collections.abc import Iterable, Set
  from typing import Any
label_desc: Use image node_id 2 for calling the JaxprInputEffect obj's underlying member method code with example usage: obj.replace() and returns: self