code: def _get_random_unit_vector(ndims, dtype):
      x = tf.random.normal((ndims,), dtype=dtype)
      return x / tf.linalg.norm(x, axis=-1, keepdims=True)
  
imports: import collections
  from tensorflow_datasets.core.utils.lazy_imports_utils import tensorflow
label_desc: Use image node_id 4 for calling a global function with example usage: _get_random_unit_vector(ndims, dtype) and returns: unknown