Change the specification of Softmax ONNX export

As @cattpku notes in issue #89, we should follow the
[PyTorch ONNX limitations](https://pytorch.org/docs/stable/onnx.html#supported-operators)
specification.  For image classifiers this change is only syntactical.