class_docstr: Wrapper for tf.data.Dataset to permit extensions.
class_docstr_tok: ["Wrapper","for","tf.data.Dataset","to","permit","extensions","."]
code: def nchw(self):
      return self.map(
          lambda x: dict(
              image=tf.transpose(x["image"], [0, 3, 1, 2]),
              label=x["label"],
          )
      )
  
file_path: objax/examples/image_classification/tfdata/data.py
imports: from typing import Callable, Optional, Tuple, List
  import numpy
  import tensorflow
label_desc: Use image node_id 8 for calling the DataSet obj's underlying member method code with example usage: obj.nchw() and returns: self