RuntimeError: shape '[784]' is invalid for input of size 7840

Question

Below is the code:

import torch

import torchvision

from torchvision import transforms, datasets

#Establishing the batch size