Could you post a code snippet showing this behavior?
In case you are using matplotlib, it should allow you to visualize images containing only the spatial size as seen here:
mask = torch.randint(0, 2, (244, 244))
plt.imshow(mask.numpy())