class_docstr: Softmax on multiple dimensions
  
  Parameters
  ----------
  axis : the axis for softmax normalization
class_docstr_tok: ["Softmax","on","multiple","dimensions","Parameters","--","--","--","--","--","axis",":","the","axis","for","softmax","normalization"]
code: def __init__(self, axis=(2, 3), **kwargs):
      super(SoftmaxHD, self).__init__(**kwargs)
      self.axis = axis
  
imports: from __future__ import absolute_import
  from mxnet import initializer
  from mxnet.gluon import nn, contrib
  from mxnet.gluon.nn import BatchNorm, HybridBlock
label_desc: Use image node_id 1 to create a new SoftmaxHD object from inherited base classes: HybridBlock with example: obj = SoftmaxHD(axis)