code: def _segment_mean(data, segment_ids, name=None):
      sm = segment_sum(data, segment_ids, name=name)
      denom = bincount(segment_ids - segment_ids[:1])
      return sm / denom.reshape(denom.shape + (1,) * (data.ndim - 1))
  
imports: import collections
  import functools
  import numpy
  import numpy
  from tensorflow_probability.python.internal.backend.numpy import _utils
  from tensorflow_probability.python.internal.backend.numpy.numpy_array import _reverse
  from tensorflow_probability.python.internal.backend.numpy.numpy_array import one_hot
  from tensorflow_probability.python.internal.backend.numpy.ops import _convert_to_tensor
label_desc: Use image node_id 17 for calling a global function with example usage: _segment_mean(data, segment_ids, name) and returns: unknown