code: def make_wishart(self, dims, new_batch_shape, old_batch_shape):
      new_batch_shape_ph = (
          tf.constant(np.int32(new_batch_shape))
          if self.is_static_shape
          else tf1.placeholder_with_default(
              np.int32(new_batch_shape), shape=None
          )
      )
  
      scale = self.dtype(
          [
              [[1.0, 0.5], [0.5, 1.0]],
              [[0.5, 0.25], [0.25, 0.75]],
          ]
      )
      scale = np.reshape(
          np.concatenate([scale, scale], axis=0),
          old_batch_shape + [dims, dims],
      )
      scale_ph = tf1.placeholder_with_default(
          scale, shape=scale.shape if self.is_static_shape else None
      )
      wishart = wishart_lib.WishartTriL(
          df=5,
          scale_tril=DeferredTensor(scale_ph, tf.linalg.cholesky),
          validate_args=True,
      )
      reshape_wishart = batch_reshape.BatchReshape(
          distribution=wishart,
          batch_shape=new_batch_shape_ph,
          validate_args=True,
      )
  
      return wishart, reshape_wishart
  
imports: import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.bijectors import bijector_test_util
  from tensorflow_probability.python.distributions import batch_reshape
  from tensorflow_probability.python.distributions import mvn_diag
  from tensorflow_probability.python.distributions import normal
  from tensorflow_probability.python.distributions import poisson
  from tensorflow_probability.python.distributions import triangular
  from tensorflow_probability.python.distributions import uniform
  from tensorflow_probability.python.distributions import wishart
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.util.deferred_tensor import DeferredTensor
label_desc: Use image node_id 1 for calling the _BatchReshapeTest obj's underlying member method code with example usage: obj.make_wishart(dims, new_batch_shape, old_batch_shape) and returns: wishart,  reshape_wishart