My code snippet is initializing a random tensor, so the output is expected.
Replace predictions
with your tensors.
@ptrblck thanks for reply, but I want to display the image of coordinates x it is the pixels of the image and y is the result of the prediction
@ptrblck @Sayed_Nadim how I get the result as an image, with x is the pixel val-x and co-ordinate is the prediction?