class_docstr: Base class of a step decoder
  
  You may inherit `BaseStepDecoder` and implement the required methods.
class_docstr_tok: ["Base","class","of","a","step","decoder","You","may","inherit","`","BaseStepDecoder","`","and","implement","the","required","methods","."]
docstr: Batch axis of the state
  
  i --> axis of the batch dimension
  None --> no batch axis in the state
docstr_tok: ["Batch","axis","of","the","state","i","--",">","axis","of","the","batch","dimension","None","--",">","no","batch","axis","in","the","state"]
code: def state_batch_axis(self):
      """Batch axis of the state
  
      i --> axis of the batch dimension
      None --> no batch axis in the state
      """
      raise NotImplementedError()
  
imports: import warnings
  import mxnet
  import abc
  from mxnet.gluon import HybridBlock
  from typing import Callable, Optional
  from .layers import get_activation
label_desc: Use image node_id 1 for calling the BaseStepDecoder obj's underlying member method code with example usage: obj.state_batch_axis() without return types