Dimension mismatch between data and nn.Linear tensors?

[ptrblck]

This topic might have the answer.