handreyeg

I'm using the PytorchTensorflowMnist.ipynb notebook available [here](https://github.com/onnx/tutorials/blob/master/tutorials/PytorchTensorflowMnist.ipynb). When I run the code below, the prediction is right (the number in the image is 2), but shouldn't the `output` variable be...