code: def reduce_mean(
      input_tensor: ragged_tensor.Ragged,
      axis=None,
      keepdims=None,
      name=None,
  ):
      """For docs, see: _RAGGED_REDUCE_DOCSTRING."""
      with ops.name_scope(
          name, "RaggedReduceMean", [input_tensor, axis]
      ):
          total = reduce_sum(input_tensor, axis, keepdims)
          if ragged_tensor.is_ragged(input_tensor):
              ones = ragged_tensor.RaggedTensor.from_nested_row_splits(
                  array_ops.ones_like(input_tensor.flat_values),
                  input_tensor.nested_row_splits,
                  validate=False,
              )
          else:
              ones = array_ops.ones_like(input_tensor)
          count = reduce_sum(ones, axis, keepdims)
          if ragged_tensor.is_ragged(total):
              return ragged_tensor.RaggedTensor.from_nested_row_splits(
                  total.flat_values / count.flat_values,
                  total.nested_row_splits,
                  validate=False,
              )
          else:
              return total / count
  
imports: import functools
  import typing
  import numpy
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import tensor
  from tensorflow.python.framework import tensor_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import check_ops
  from tensorflow.python.ops import gen_ragged_math_ops
  from tensorflow.python.ops import map_fn
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import nn_ops
  from tensorflow.python.ops.ragged import ragged_functional_ops
  from tensorflow.python.ops.ragged import ragged_tensor
  from tensorflow.python.ops.ragged import segment_id_ops
  from tensorflow.python.util import dispatch
  from tensorflow.python.util.tf_export import tf_export
label_desc: Use image node_id 16 for calling a global function with example usage: reduce_mean(input_tensor, axis, keepdims, name) and returns: ragged_tensor, unknown