MarcSteven:

Based on the demo apple offered the model deal with the softmax layer and then only show the result.


I was referring to this demo you mentioned.
Based on the graph you’ve posted I assume it’s a CoreML graph, as these operations do not seem to come from PyTorch directly.
Could you explain your use case completely, i.e. what kind of model or code are you using now and what are you trying to achieve?