class_docstr: Wrapper for tf.data.Dataset to permit extensions.
class_docstr_tok: ["Wrapper","for","tf.data.Dataset","to","permit","extensions","."]
code: def one_hot(self, nclass):
      return self.dmap(
          lambda label, **kw: dict(
              label=tf.one_hot(label, nclass), **kw
          )
      )
  
imports: import os
  from typing import Callable, Optional, Tuple, List, Iterable
  import numpy
  import tensorflow
  from absl import flags
label_desc: Use image node_id 9 for calling the DataSet obj's underlying member method code with example usage: obj.one_hot(nclass) and returns: self