That one means you are displaying your matrix in a higher dimensional space.
PyTorch usually dedicates one dimension to the batch size.
You can fix it by slicing your tensor
input[:1]
or by adding an aditional dimension to the GT
target.unsqueeze(1)