I am trying to implement one-hot encoding for MNIST imported from Kaggle. The shape of the encoding is [1, 10] but when the loss function runs, it throws the following error:
ValueError: Expected input batch_size (10) to match target batch_size (256).
My mini batch-size is 256.
What should I do?