I don’t need the actual data, just the shapes. 
In my example I’m using x = torch.randn(200, 3087), which is clearly wrong and is raising the shape mismatch, so which shapes do the inputs have in your code?