I am getting a runtime error of tensor size not matching while running the test.py file on google colab. Please help. 
I am getting a runtime error of tensor size not matching while running the test.py file on google colab. Please help.