The dimension is wrong for a nn.Conv2d layer.
Do you have 2 images? If so, you could remove dimension 0 with batch_x.squeeze_(0).
Could you post your model architecture, so that I can have a look, if this is an internal bug?