Mismatch of output dimensions and whats reported from model summary in pytorch

Question

I am struggeling to determine a cause for following behavior of 3 class classification code I have below. The classifier takes a 2D matrix of torch.Size([1, 1, 512, 512]) .