class_docstr: Mimics a user-defined bijector that is registered as a Pytree.
class_docstr_tok: ["Mimics","a","user-defined","bijector","that","is","registered","as","a","Pytree","."]
code: def __init__(self, shift):
      parameters = dict(locals())
      self.shift = shift
      super(PytreeShift, self).__init__(
          validate_args=True,
          forward_min_event_ndims=0,
          parameters=parameters,
          name="pytree_shift",
      )
  
imports: from absl import logging
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import bijector
  from tensorflow_probability.python.bijectors import reshape
  from tensorflow_probability.python.distributions import uniform
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math.gradient import batch_jacobian
label_desc: Use image node_id 1 to create a new PytreeShift object from inherited base classes: bijector_lib with example: obj = PytreeShift(shift)