How can I make a transformation that results in a sequential MNIST dataset? I cannot seem to figure this out. .view(-1, 784) seems like it would work, but does seem to be correct and I get the error:
RuntimeError: size mismatch, m1: [16 x 784], m2: [28 x 400]. If I do .view(-1, 784).t() the program doesn’t have the error but the network doesn’t learn anything. This is with batch_first being set to True.