Cannot understand issue! Input tensor must be at least 2D

Question

I have written this simple program to make a prediction (do not mind that there is no train/test split). x is a 2D input array of size 40k, 4.