TypeError: argument of type 'Tensor' is not iterable

Question

I am getting this error while trying to make prediction on test data from my model. Code:

x_train,x_test,y_train,y_test = train_test_split(x, y, test_size=0.1