CUDA error: an illegal memory access was encountered

Question

I am unable to do device conversion for a PyTorch tensor on Google Colab. I attach the code snippet below:
Please help me.