code: def broadcast_compatible_shape(shape):
      """Strategy for drawing shapes broadcast-compatible with `shape`."""
      # broadcasting_shapes draws a sequence of shapes, so that the last "completes"
      # the broadcast to fill out batch_shape. Here we just draw two and take the
      # first (incomplete) one.
      return broadcasting_shapes(shape, 2).map(lambda shapes: shapes[0])
  
imports: import contextlib
  import os
  import re
  import traceback
  import unittest
  import hypothesis
  from hypothesis.extra import numpy
  import hypothesis.strategies
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.util.deferred_tensor import DeferredTensor
label_desc: Use image node_id 26 for calling a global function with example usage: broadcast_compatible_shape(shape) and returns: broadcasting_shapes