code: def _take(
      self, count, name=None
  ):  # pylint: disable=unused-private-name
      return _TakeDataset(self, count, name=name)
  
imports: from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import gen_dataset_ops
label_desc: Use image node_id 1 for calling a global function with example usage: _take(count, name) and returns: _TakeDataset