I did as you said but i am getting the following error.
TypeError: expected Tensor as element 0 in argument 0, but got tuple