Hey! Thank you for responding! My output is of the shape [100, 11, 80, 80] where batch = 100, n_classes = 11, height/width = 80. How do I convert this tensor to a tensor that will be accepted by cross entropy loss?