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’