Solved by ntreepoint in post #2 
              
              
                I got the same issue. 
Using a newer version of Pytorch mobile has solved the issue. 
Here is what i included in build.gradle: 
implementation ‘org.pytorch:pytorch_android:1.5.0’ 
implementation ‘org.pytorch:pytorch_android_torchvision:1.5.0’