code: def data_batch_axis(self):
      return 0 if self._layout == "NT" else 1
  
file_path: gluon-nlp/docs/model_zoo/generation/interactive_conditional_gpt2_samples.py
imports: import os
  import mxnet
  import argparse
  from gluonnlp.utils import set_seed
  from gluonnlp.sequence_sampler import BeamSearchSampler, BaseStepDecoder
  from gluonnlp.models.gpt2 import GPT2ForLM, list_pretrained_gpt2, get_pretrained_gpt2
label_desc: Use image node_id 3 for calling the GPT2Decoder obj's underlying member method code with example usage: obj.data_batch_axis() and returns: unknown