code: def _bcast_shape(base_shape, args):
      bcast_shape = _ensure_shape_tuple(base_shape)
      for arg in args:
          bcast_shape = ops.broadcast_shape(
              bcast_shape, np.asarray(arg).shape
          )
      return bcast_shape
  
imports: import numpy
  import numpy
  from tensorflow_probability.python.internal.backend.numpy import _utils
  from tensorflow_probability.python.internal.backend.numpy import ops
  from tensorflow_probability.python.internal.backend.numpy.numpy_math import softmax
label_desc: Use image node_id 2 for calling a global function with example usage: _bcast_shape(base_shape, args) and returns: bcast_shape