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.

def convert_softmax(net, node, module, builder):
    """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.
    """
    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)