You would have to check why these tensors have different shapes, which could be caused by an error in the model architecture, a typo using the wrong variable name, etc.
Based on the shapes it seems as if output might have the same dimensions as an image tensor input.