Output and Broadcast shape mismatch in MNIST, torchvision and getting dead kernel if i fix it by changing transform

Question

I am getting following error when using MNIST dataset in Torchvision

output with shape [1, 28, 28] doesn\'t match the broadcast shape [3, 28, 28]

And when Im d