code: def convert_softmax(net, node, module, builder):
      
      input_name, output_name = _get_input_output_name(net, node)
      name = node['name']
      param = _get_attr(node)
  
      if param is not None and 'axis' in param:
          axis = literal_eval(param['axis'])
          assert axis == 1, "Only softmax with axis 1 is supported"
  
      builder.add_softmax(name=name,
                          input_name=input_name,
                          output_name=output_name)
docs: Convert a softmax layer from mxnet to coreml.
  
      Parameters
      ----------
      net: network
          A mxnet network object.
  
      node: layer
          Node to convert.
  
      module: module
          An module for MXNet
  
      builder: NeuralNetworkBuilder
          A neural network builder object.