MyModelA uses a nn.Conv2d layer as the first layer, which expects a 4-dimensional input with the shape [batch_size, channels, height, width].
In your code snippet, you are defining x1 as a 2-dimensional tensor, which will throw this error.
Based on the shape, it seems as if you would only want to use self.fc on this input?