Any simplest PyTorch MNIST examples only use matrix

this is an example for testing concept.